Developer Documentation

The developer documentation provides deeper insight to TORAX code structure, underlying libraries, and API.

• TORAX code structure
• The 1D finite volume element (fvm) library
• How to integrate new models
• Designing JAX-compatible classes
• Running a completely JITted simulation
• Geometry overrides in TORAX