maxencefaldor / cax
Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
☆180Updated 2 weeks ago
Alternatives and similar repositories for cax:
Users that are interested in cax are comparing it to the libraries listed below
- Curiosity driven exploration of your complex system 👀☆37Updated 3 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆82Updated 2 months ago
- An Open-Ended Agentic Simulator☆48Updated 9 months ago
- ☆146Updated last month
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆178Updated last month
- ☆39Updated 3 years ago
- Efficient baselines for autocurricula in JAX.☆187Updated 8 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆205Updated last month
- ☆77Updated last month
- Graph neural networks in JAX.☆67Updated 10 months ago
- Accelerated minigrid environments with JAX☆135Updated 9 months ago
- ☆37Updated this week
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- ☆166Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆312Updated 2 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆98Updated 7 months ago
- ☆69Updated 3 weeks ago
- JAX Arrays for human consumption☆92Updated last year
- Hierarchical Associative Memory User Experience☆101Updated last year
- Tools for working with the Abstraction & Reasoning Corpus☆187Updated 9 months ago
- fast + parallel AlphaZero in JAX☆96Updated 4 months ago
- 🧱 Modula software package☆188Updated last month
- Highly scalable 2D JAX physics engine.☆56Updated 2 months ago
- Automating the Search for Artificial Life with Foundation Models!☆409Updated 3 months ago
- A functional training loops library for JAX☆87Updated last year
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 9 months ago
- Pytorch-like dataloaders for JAX.☆81Updated last week
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆71Updated last year
- Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).☆22Updated last month