Nano Optax is a compact optimization library built around JAX. It is a place for experimenting with JIT-compiled first- and second-order optimization algorithms while keeping the implementation small enough to inspect.