RobertTLange / evosaxLinks
Evolution Strategies in JAX đĻ
â713Updated 3 months ago
Alternatives and similar repositories for evosax
Users that are interested in evosax are comparing it to the libraries listed below
Sorting:
- Accelerated Quality-Diversityâ334Updated 2 months ago
- RL Environments in JAX đâ846Updated 7 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ794Updated last month
- â929Updated last year
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ380Updated 2 months ago
- Really Fast End-to-End Jax RL Implementationsâ1,010Updated last year
- A bare-bones Python library for quality diversity optimization.â249Updated 3 weeks ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ268Updated 3 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â254Updated 2 months ago
- â613Updated last month
- âī¸ Vectorized RL game environments in JAXâ567Updated 10 months ago
- â913Updated 2 weeks ago
- Modular framework for Reinforcement Learning in pythonâ181Updated 2 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â740Updated 3 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ546Updated 2 weeks ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â864Updated last year
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.â1,112Updated last month
- Accelerated minigrid environments with JAXâ154Updated 2 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â362Updated 6 months ago
- Multi-Agent Reinforcement Learning with JAXâ714Updated 3 weeks ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/â524Updated 2 weeks ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.â1,013Updated 3 weeks ago
- A JAX-based framework for genetic programmingâ30Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.â303Updated 3 weeks ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.â622Updated this week
- â1,399Updated last month
- map elites python reference implementationâ96Updated 2 years ago
- Distributed Reinforcement Learning accelerated by Lightning Fabricâ416Updated 2 weeks ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â321Updated 3 weeks ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.â666Updated 4 months ago