maxencefaldor / cax
Cellular Automata Accelerated in JAX (Oral at ICLR 2025).
☆81Updated 3 months 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 👀☆35Updated 3 weeks ago
- ☆37Updated 2 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆53Updated 2 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 3 weeks ago
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- ☆70Updated 5 months ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Flow-matching algorithms in JAX☆83Updated 6 months ago
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).☆17Updated 7 months ago
- OMNI: Open-endedness via Models of human Notions of Interestingness☆42Updated 3 weeks ago
- Gradient-based constrained optimization for JAX☆29Updated 2 years ago
- ☆40Updated 2 months ago
- Code for "Goal-Guided Neural Cellular Automata: Learning to Control Self-Organising Systems"☆55Updated 2 years ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆40Updated last month
- ☆59Updated 2 years ago
- Visualizations of the theory behind diffusion models.☆78Updated 10 months ago
- OpenAI gym environment for evolving morphologies of 2D virtual creatures.☆33Updated last year
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- 3D Gaussian Splatting in JAX☆55Updated 8 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆126Updated last week
- ☆16Updated last year
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆27Updated 4 years ago
- Neural Optimal Transport with Lagrangian Costs☆52Updated 7 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- Efficient baselines for autocurricula in JAX.☆179Updated 5 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆59Updated 6 months ago