torus-solver

JAX-differentiable surface-current modeling and optimization on a circular torus, with targets motivated by stellarators and tokamaks.

This documentation is written for two audiences:

  1. Newcomers: you can start from scratch, learn the physics assumptions and equations, and run the examples end-to-end.

  2. Domain experts: you can quickly map the equations to the implementation details, understand algorithmic tradeoffs, and extend the code.