Swift implementation of the TORAX tokamak transport simulator—differentiable, MLX-powered, and tuned for Apple Silicon. - View it on GitHub