search

google-deepmind / torax

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

Tidy standard geometry code #232

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 3 months ago

Tidy standard geometry code

Remove some of the intermediate interpolation functions and use face_to_cell where possible