f0uriest

autobound

AutoBound automatically computes upper and lower bounds on functions.

pytest-testmon

Selects tests affected by changed files. Executes the right tests first. Continuous test runner when used with pytest-watch.

scalax

A simple library for scaling up JAX programs

optimagic

optimagic is a Python package for numerical optimization. It is a unified interface to optimizers from SciPy, NlOpt and other packages. optimagic's minimize function works just like SciPy's, so you don't have to adjust your code. You simply get more optimizers for free. On top you get diagnostic tools, parallel numerical derivatives and more.

testax

Jit-able runtime assertions for JAX in NumPy style.

jax-finufft

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library

lawson-criterion-paper

Progress toward fusion energy breakeven and gain as measured against the Lawson criterion, figures and data

missing-semester

The Missing Semester of Your CS Education 📚

BAD

A repo for calculating the bounce-averaged drifts

fusionrate

Fusion cross sections and rate coefficients

deepxde

A library for scientific machine learning and physics-informed learning

idrlnet

IDRLnet, a Python toolbox for modeling and solving problems through Physics-Informed Neural Network (PINN) systematically.

equinox

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

neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

quadpack

Modern Fortran QUADPACK Library for 1D numerical quadrature

optree

OpTree: Optimized PyTree Utilities

cola

Compositional Linear Algebra

alpa

Training and serving large-scale neural networks with auto parallelization.

optimistix

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

lineax

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

jax-tqdm

Add a tqdm progress bar to your JAX scans and loops.

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Tools-to-Design-or-Visualize-Architecture-of-Neural-Network

Tools to Design or Visualize Architecture of Neural Network

jaxpr-viz

Jaxpr Visualisation Tool

scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

desc_utils

vmec_equilibria

power-systems-optimization

Power systems optimization course materials

cookie

Scientific Python Library Development Guide and Cookiecutter

ripgrep

ripgrep recursively searches directories for a regex pattern while respecting your gitignore