TORAX: Google DeepMind's tokamak transport simulator.
- Extending capabilities to model spherical tokamaks
- Developing optimisation and control routines
- Integrating with and benchmarking against other codes
GPJax: Gaussian Processes in JAX/Flax.
- Improving core fitting routines and enhancing flexibility
- Developing new tools for Bayesian optimisation
Multi-objective Bayesian optimization for design of Pareto-optimal current drive profiles in STEP | IEEE Transactions on Plasma Science (2023) | π Paper β’ π» Code |
Sample-efficient Bayesian optimisation using known invariances | NeurIPS (2024) | π Paper β’ π» Code |
invariantkernels
: Group-invariant kernels for symmetry-aware GPs in GPyTorch.