API reference

The API is intentionally small and “research-friendly”: most routines are plain functions operating on jax.numpy arrays and small dataclasses.

Note

torus_solver.gui_vtk can be imported without VTK installed, but running the GUI requires the optional extra torus-solver[gui].