TORAX: Tokamak transport simulation in JAX

Theano was a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently. It is being continued as PyTensor: www.github.…

Optimizing GPU-meta-programming code generating array oriented optimizing math compiler in Python

Warning: This project does not have any current developer. See bellow.