maxencefaldor / cax
Cellular Automata Accelerated in JAX
☆70Updated 2 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for cax
- ☆36Updated 2 years ago
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 2 years ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code☆27Updated last week
- ☆48Updated 9 months ago
- Curiosity driven exploration of your complex system 👀☆27Updated last month
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Graph neural networks in JAX.☆67Updated 5 months ago
- The Energy Transformer block, in JAX☆53Updated 11 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆84Updated last month
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).☆14Updated 4 months ago
- Efficient baselines for autocurricula in JAX.☆173Updated 2 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆22Updated this week
- Flow-matching algorithms in JAX☆77Updated 3 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆56Updated 3 months ago
- ☆63Updated 3 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- A programming language for formal/informal computation.☆40Updated 5 months ago
- ☆58Updated 2 years ago
- OMNI: Open-endedness via Models of human Notions of Interestingness☆39Updated 11 months ago
- Fast singularity detection with kernel☆33Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costs☆49Updated 4 months ago
- Gradient-based constrained optimization for JAX☆26Updated 2 years ago
- An Open-Ended Agentic Simulator☆28Updated 3 months ago
- Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions"☆49Updated this week
- A functional training loops library for JAX☆85Updated 9 months ago
- OpenAI gym environment for evolving morphologies of 2D virtual creatures.☆33Updated last year
- Visualizations of the theory behind diffusion models.☆75Updated 7 months ago
- ☆50Updated 5 months ago