search

google-deepmind / torax

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

Allow interpolated var to take arrays instead of dicts #229

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Allow interpolated var to take arrays instead of dicts

This makes it possible to use it easily for the geometry