RobertTLange / evosaxLinks
Evolution Strategies in JAX đĻ
â695Updated 2 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â331Updated last month
- RL Environments in JAX đâ839Updated 6 months ago
- â926Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ785Updated last week
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ375Updated last month
- A bare-bones Python library for quality diversity optimization.â245Updated this week
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ264Updated 2 months ago
- Really Fast End-to-End Jax RL Implementationsâ989Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â250Updated last month
- â607Updated this week
- âī¸ Vectorized RL game environments in JAXâ553Updated 9 months ago
- â899Updated 3 weeks ago
- Modular framework for Reinforcement Learning in pythonâ180Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ531Updated last month
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.â1,107Updated 3 weeks ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â734Updated 3 years ago
- Accelerated minigrid environments with JAXâ153Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â356Updated 5 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â856Updated last year
- Hardware accelerated, batchable and differentiable optimizers in JAX.â1,005Updated 2 months ago
- â1,388Updated this week
- Multi-Agent Reinforcement Learning with JAXâ698Updated 2 weeks ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/â507Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.â297Updated last week
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ70Updated 3 years ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.â619Updated last week
- map elites python reference implementationâ96Updated 2 years ago
- Gaussian processes in JAX and Flax.â561Updated last week
- A JAX-based framework for genetic programmingâ29Updated last month
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.â676Updated 2 weeks ago