google-deepmind / fusion_surrogatesLinks
A library of surrogate transport models for tokamak fusion.
☆29Updated 2 months ago
Alternatives and similar repositories for fusion_surrogates
Users that are interested in fusion_surrogates are comparing it to the libraries listed below
Sorting:
- TORAX: Tokamak transport simulation in JAX☆616Updated this week
- Tokamak plasma operation control through multi-objective reinforcement learning in KSTAR☆25Updated 11 months ago
- A tokamak (nuclear fusion reactor) simulator with LSTM-based neural network (KSTAR-NN)☆87Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆509Updated last week
- Implementation of Forward Laplacian algorithm in JAX☆94Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Updated this week
- AI design of tokamak operation for autonomous control of fusion plasma.☆30Updated 3 years ago
- Additive manufacturing simulation with JAX.☆339Updated 6 months ago
- ☆15Updated 3 weeks ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated this week
- From-scratch C++ and Python reimplementation of the Variational Moments Equilibrium Code (VMEC).☆206Updated last week
- Jax-based quantum state vector simulator.☆25Updated 7 months ago
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆92Updated this week
- Interpolation and function approximation with JAX☆248Updated last week
- Orthogonal polynomials with JAX☆26Updated 2 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated 2 weeks ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Updated 2 years ago
- Python implementation of the Quasisymmetric Stellarator Construction method☆16Updated last year
- A framework for creating databases for surrogate models of complex physics codes.☆12Updated this week
- Bluemira is an integrated inter-disciplinary design tool for future fusion reactors. It incorporates several modules, some of which rely …☆95Updated last week
- Simons Stellarator Optimizer Code☆137Updated last week
- Automatic Functional Differentiation in JAX☆81Updated 4 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆438Updated 3 weeks ago
- Multiple dispatch over abstract array types in JAX.☆136Updated last month
- JAX-DIPS is a differentiable interfacial PDE solver.☆49Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week
- A curated list of awesome Scientific Machine Learning (SciML) papers, resources and software☆78Updated 2 years ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆48Updated 2 years ago
- ☆16Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆536Updated last month