Kernel dies when training simple SCVI on mac m3

I’m facing dying kernel on mac os (Sanorama 14.1.2 (23B2091)), chip M3 with 36GB memory. I have explored discussions on this matter but none of them work.
I have create an environment using mamba to have native python but it didn’t resolved the issue. I’m using other packages in my code. An export of the env is accessible via this link.

Any help highly appreciated in advance

@adamgayoso
Hey Adam! do you have any idea about this issue?