Framework to implement JAX models that can be distributed over MPI
Primary LanguagePythonMIT LicenseMIT
No one’s star this repository yet.