How to create rational LLM-based agents? Using game-theoretic workflows!

Optax is a gradient processing and optimization library for JAX.

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

A module to provide the numerical inverse of functions