google-deepmind / torax

TORAX: Tokamak transport simulation in JAX
https://torax.readthedocs.io
Other
345 stars 31 forks source link

Move the mesh into the intermediates #227

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 3 months ago

Move the mesh into the intermediates

This doesn't change across interpolated values and it allow us to make that clear/use it earlier for radial interpolation