SFINCS: the Stellarator Fokker-Planck Iterative Neoclassical Conservative Solver

Serial VMEC for educational purposes (e.g. understanding the algorithms etc.)

Interpolation and function approximation with JAX

Playing Pokemon Red with Reinforcement Learning

High-Performance Symbolic Regression in Python and Julia

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

A list of manuscripts/tools using diffusion on biological enttieis

JAX Synergistic Memory Inspector

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

The Mojo Programming Language

AutoBound automatically computes upper and lower bounds on functions.

Physical Symbolic Optimization

Monte Carlo tree search in JAX

A playbook for systematically maximizing the performance of deep learning models.

3D Equilibrium Solver

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

PyTorch implementation of Neural Netowrk Differential Equation Plasma Equilibrium Solver.

Config files for my GitHub profile.

Convert Machine Learning Code Between Frameworks

What the f*ck Python? 😱

Model interpretability and understanding for PyTorch

Stellarator Equilibrium and Optimization Suite

A collection of handy Bash One-Liners and terminal tricks for data processing and Linux system maintenance.

Memray is a memory profiler for Python

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.