maxencefaldor / cax
Cellular Automata Accelerated in JAX
☆68Updated last week
Related projects ⓘ
Alternatives and complementary repositories for cax
- ☆36Updated 2 years ago
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Flow-matching algorithms in JAX☆74Updated 3 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- 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☆26Updated 2 months ago
- ☆58Updated 2 years ago
- Visualizations of the theory behind diffusion models.☆74Updated 6 months ago
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- ☆81Updated this week
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated last year
- ☆48Updated 8 months ago
- ☆40Updated 4 months ago
- Code for minimum-entropy coupling.☆29Updated 4 months ago
- Efficient baselines for autocurricula in JAX.☆173Updated 2 months ago
- Repository for "Toward Artificial Open-Ended Evolution within Lenia using Quality-Diversity" (ALIFE 2024).☆14Updated 3 months ago
- The Energy Transformer block, in JAX☆50Updated 10 months ago
- Curiosity driven exploration of your complex system 👀☆27Updated 3 weeks ago
- Graph neural networks in JAX.☆67Updated 4 months ago
- ☆61Updated 2 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated last month
- Neural Optimal Transport with Lagrangian Costs☆47Updated 3 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆21Updated 4 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆84Updated last month
- fast + parallel AlphaZero in JAX☆84Updated 7 months ago
- σ-GPT: A New Approach to Autoregressive Models☆59Updated 2 months ago
- Jax like function transformation engine but micro, microjax☆26Updated 2 weeks ago
- 3D Gaussian Splatting in JAX☆54Updated 5 months ago
- Pytorch-like dataloaders in JAX.☆59Updated 3 weeks ago
- Fast singularity detection with kernel☆33Updated 10 months ago