maxencefaldor / caxLinks
Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
☆205Updated 2 months ago
Alternatives and similar repositories for cax
Users that are interested in cax are comparing it to the libraries listed below
Sorting:
- Curiosity driven exploration of your complex system 👀☆40Updated 5 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆91Updated 4 months ago
- ☆164Updated 3 months ago
- ☆55Updated last year
- Automating the Search for Artificial Life with Foundation Models!☆425Updated 6 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆171Updated 2 years ago
- Efficient baselines for autocurricula in JAX.☆189Updated 10 months ago
- The boundary of neural network trainability is fractal☆210Updated last year
- Flexible Inference for Predictive Coding Networks in JAX.☆51Updated 2 weeks ago
- ☆39Updated 3 years ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆194Updated last week
- Diffusion model derived evolutionary algorithm☆214Updated last month
- ☆40Updated 2 months ago
- Brain-like variational inference☆55Updated last month
- Tools for working with the Abstraction & Reasoning Corpus☆194Updated 11 months ago
- ☆135Updated 2 months ago
- Graph neural networks in JAX.☆67Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆60Updated 6 months ago
- 🧱 Modula software package☆204Updated 3 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated 11 months ago
- ☆273Updated last year
- An Open-Ended Agentic Simulator☆51Updated 11 months ago
- A functional training loops library for JAX☆88Updated last year
- Run PyTorch in JAX. 🤝☆256Updated 2 weeks ago
- Getting crystal-like representations with harmonic loss☆191Updated 3 months ago
- Evolution Strategies in JAX 🦎☆641Updated last month
- fast + parallel AlphaZero in JAX☆97Updated 6 months ago
- Compositional Linear Algebra☆478Updated last month
- Scalable and Stable Parallelization of Nonlinear RNNS☆17Updated 5 months ago