/multigrad

Framework to implement JAX models that can be distributed over MPI

Primary LanguagePythonMIT LicenseMIT

multigrad

Framework to implement JAX models that can be distributed over MPI

Author

  • Alan Pearl

Documentation

Online documentation is available at multigrad.readthedocs.io.