Enable multi-gpu batching via array sharding

facebookresearch / fmmax

Fourier modal method with Jax

MIT License

95 stars 9 forks source link

Closed smartalecH closed 2 weeks ago

mfschubert commented 1 month ago

@smartalecH Should we close this? I guess it's no longer needed?

smartalecH commented 2 weeks ago

Ya let's close for now, and I can revive it when needed.