TORAX: Tokamak transport simulation in JAX
☆625Updated this week
Alternatives and similar repositories for torax
Users that are interested in torax are comparing it to the libraries listed below
Sorting:
- FreeGSNKE: A Python code for evolutive free-boundary tokamak plasma equilibrium simulations☆59Updated this week
- Tokamak plasma operation control through multi-objective reinforcement learning in KSTAR☆26Mar 7, 2025Updated 11 months ago
- A tokamak (nuclear fusion reactor) simulator with LSTM-based neural network (KSTAR-NN)☆88Oct 8, 2023Updated 2 years ago
- Bluemira is an integrated inter-disciplinary design tool for future fusion reactors. It incorporates several modules, some of which rely …☆96Feb 23, 2026Updated last week
- Routines for tokamak analysis☆23Aug 21, 2018Updated 7 years ago
- Free boundary Grad-Shafranov solver☆127Dec 30, 2025Updated 2 months ago
- AI design of tokamak operation for autonomous control of fusion plasma.☆30Jul 6, 2022Updated 3 years ago
- A library of surrogate transport models for tokamak fusion.☆30Nov 13, 2025Updated 3 months ago
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- The Open FUSION Toolkit (OFT) is a suite of modeling tools, and their underlying finite element framework, for problems in plasma and fus…☆58Feb 21, 2026Updated last week
- FUsion Synthesis Engine☆71Updated this week
- An open-source physics-based Scientific Framework for Disruption Analysis of Fusion Plasmas for AI/ML applications☆33Updated this week
- PROCESS is a systems code at UKAEA that calculates in a self-consistent manner the parameters of a fusion power plant with a specified pe…☆58Updated this week
- Generic Data Model for experimental and simulation fusion data☆57Feb 19, 2026Updated last week
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,912Feb 23, 2026Updated last week
- Stellarator Equilibrium and Optimization Suite☆148Updated this week
- Reference implementation for the Hasegawa-Wakatani model of plasma turbulence inside nuclear fusion reactors in two dimensions☆31Aug 17, 2024Updated last year
- Computational Fluid Dynamics in JAX☆927Jan 14, 2026Updated last month
- List of interesting fusion open source projects☆33Feb 13, 2026Updated 2 weeks ago
- PPPL deep learning disruption prediction package☆88Mar 13, 2023Updated 2 years ago
- Free boundary Grad-Shafranov solver for time evolution☆12Feb 13, 2026Updated 2 weeks ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆516Feb 23, 2026Updated last week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- Gaussian processes in JAX and Flax.☆593Feb 20, 2026Updated last week
- A collection of files that can be used as inputs in the codes commonly used in the Plasma Physics community.☆50Oct 3, 2024Updated last year
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆516Feb 22, 2026Updated last week
- Variational Autoencoder (VAE)-like neural network to solve ideal MHD equilibrium in a tokamak☆11May 20, 2022Updated 3 years ago
- ☆16Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆483Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆120Feb 3, 2026Updated 3 weeks ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆35Updated this week
- ☆19Mar 24, 2023Updated 2 years ago
- JAX - A curated list of resources https://github.com/google/jax☆2,058Jan 20, 2026Updated last month
- Read and write the popular "geqdsk" and "aeqdsk" tokamak equilibrium file formats☆18Feb 20, 2025Updated last year
- Turn SymPy expressions into trainable JAX expressions.☆362Feb 4, 2026Updated 3 weeks ago
- An open source Python package for plasma research and education☆657Updated this week
- List of awesome university courses to learn Plasma Physics and Fusion!☆72Jul 27, 2023Updated 2 years ago