foragax

github.comtool

Agent-Based modelling framework in JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Pax github.com

   tool category #jax link ↗

   A Jax-based machine learning framework for training large scale models.

2. ▲
   evosax github.com

   tool category #jax link ↗

   JAX-Based Evolution Strategies

   
3. ▲
   RLax github.com

   tool category #jax link ↗

   Library for implementing reinforcement learning agents.

4. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   
5. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   
6. ▲
   efax github.com

   tool category #jax link ↗

   Exponential Families in JAX.

   

JAX › Libraries > New Libraries: “Agent-Based modelling framework in JAX.”