RobertTLange / evosax
Evolution Strategies in JAX đĻ
â550Updated last week
Alternatives and similar repositories for evosax:
Users that are interested in evosax are comparing it to the libraries listed below
- RL Environments in JAX đâ703Updated 7 months ago
- Accelerated Quality-Diversityâ275Updated this week
- â873Updated 7 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ685Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementationsâ810Updated 5 months ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ225Updated 3 weeks ago
- A bare-bones Python library for quality diversity optimization.â217Updated last month
- Modular framework for Reinforcement Learning in pythonâ171Updated 2 years ago
- â188Updated 2 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ277Updated this week
- âī¸ Vectorized RL game environments in JAXâ440Updated this week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â801Updated 6 months ago
- â823Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â654Updated 2 years ago
- â548Updated 2 months ago
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.â1,038Updated this week
- â1,288Updated 3 weeks ago
- Multi-Agent Reinforcement Learning with JAXâ509Updated last week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ396Updated this week
- GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)â100Updated this week
- Accelerated minigrid environments with JAXâ128Updated 6 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/â377Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â278Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.â263Updated 2 weeks ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.â947Updated 5 months ago
- A suite of test scenarios for multi-agent reinforcement learning.â649Updated last week
- â295Updated 2 months ago
- map elites python reference implementationâ80Updated last year
- â153Updated last year
- Multi-objective Gymnasium environments for reinforcement learningâ306Updated 2 weeks ago