surana01's Stars
nomadic-ml/nomadic
Nomadic is an enterprise-grade framework for teams to continuously optimize compound AI systems
EdanToledo/Stoix
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
ScottAlexanderCameron/Jynx
A neural network library written in jax
instadeepai/compass
:compass: COMPASS: Combinatorial Optimization with Policy Adaptation using Latent Space Search
instadeepai/flashbax
⚡ Flashbax: Accelerated Replay Buffers in JAX
instadeepai/jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX