Welcome to multigrad’s documentation!

multigrad allows you to easily implement data-parallelized (over MPI), differentiable models in the Jax framework. The code is open-source and available on GitHub.

Contents:

• Installation Instructions
  • Installation
  • Prerequisites
• Quickstart Tutorial
  • Example Model
  • Inspecting the “truth”
  • Introducing MultiGrad
  • Recovering the truth with gradient descent
  • The same thing, but parallel!
• API Reference
  • multigrad

Indices and tables

• Index

• Search Page