Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
☆246Nov 24, 2025Updated 3 months ago
Alternatives and similar repositories for cax
Users that are interested in cax are comparing it to the libraries listed below
Sorting:
- Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).☆26Apr 2, 2025Updated 11 months ago
- ☆20Feb 8, 2024Updated 2 years ago
- Curiosity driven exploration of your complex system 👀☆43Jan 30, 2025Updated last year
- ☆41Apr 27, 2022Updated 3 years ago
- A list of paper and resources regarding Neural Cellular Automata☆22Feb 5, 2026Updated last month
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆233Feb 26, 2026Updated last week
- ☆55May 24, 2024Updated last year
- Evolution Strategies in JAX 🦎☆734Sep 20, 2025Updated 5 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆374Feb 10, 2026Updated 3 weeks ago
- Efficient baselines for autocurricula in JAX.☆208Aug 24, 2024Updated last year
- Generative cellular automaton-like learning environments for RL.☆20Jan 30, 2025Updated last year
- Simple implementations of Cartesian Genetic Programming (CGP) and Linear Genetic Programming (LGP) in JAX☆11Oct 28, 2025Updated 4 months ago
- Accelerated Quality-Diversity☆340Oct 30, 2025Updated 4 months ago
- Automating the Search for Artificial Life with Foundation Models!☆451Oct 23, 2025Updated 4 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Aug 18, 2024Updated last year
- Diffusion model derived evolutionary algorithm☆246Jun 17, 2025Updated 8 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Updated this week
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆28Jul 11, 2024Updated last year
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆35Oct 3, 2025Updated 5 months ago
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆108Nov 25, 2025Updated 3 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithms☆200Dec 19, 2025Updated 2 months ago
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆20Dec 1, 2025Updated 3 months ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆325Dec 16, 2025Updated 2 months ago
- Deep Networks Grok All the Time and Here is Why☆38May 18, 2024Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 months ago
- ☆399Jan 9, 2026Updated last month
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆101Jan 23, 2025Updated last year
- ☆69Mar 1, 2025Updated last year
- ☆92Feb 16, 2026Updated 2 weeks ago
- Fast singularity detection with kernel☆38Jan 4, 2024Updated 2 years ago
- ☆20Mar 4, 2025Updated last year
- ☆22May 12, 2025Updated 9 months ago
- ☆35Nov 22, 2024Updated last year
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 3 months ago
- ☆93Jan 21, 2026Updated last month
- RL Environments in JAX 🌍☆868May 30, 2025Updated 9 months ago
- ☆52Nov 6, 2025Updated 4 months ago