Adding this className to the DataGrid sx allows you to manipulate the individual headers but not the whole container. You can still set background colors and margins as you want.

'& .MuiDataGrid-columnHeader': {
                    backgroundColor: 'your-color-constant',
                }