mzguntalan / zephyrLinks
Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulation
☆36Updated 3 weeks ago
Alternatives and similar repositories for zephyr
Users that are interested in zephyr are comparing it to the libraries listed below
Sorting:
- ☆27Updated last year
- A pure and fast NumPy implementation of Mamba with cache support.☆17Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆91Updated 4 months ago
- An implementation of AlphaZero and MCTS with neural networks for Tetris☆21Updated 3 months ago
- Schedule free optimiser implemented in JAX using Optimistix☆14Updated last year
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆206Updated 2 months ago
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated 2 months ago
- 🧱 Modula software package☆204Updated 3 months ago
- coloring terminal text with intensities (used for plotting probability, entropy with tokens)☆12Updated 9 months ago
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆97Updated 6 months ago
- Tensor library with autograd using only Rust's standard library☆68Updated last year
- Cost aware hyperparameter tuning algorithm☆162Updated last year
- Training AI for Super Smash Bros. Melee☆27Updated 3 months ago
- Code for the Fractured Entangled Representation Hypothesis position paper!☆135Updated last month
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆60Updated 6 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆60Updated 9 months ago
- nanoGPT using Equinox☆13Updated 2 years ago
- A reinforcement learning codebase focusing on the emergence of cooperation and alignment in multi-agent AI systems.☆63Updated this week
- ☆43Updated last month
- An implementation of PSGD Kron second-order optimizer for PyTorch☆92Updated 3 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆68Updated 2 months ago
- fast + parallel AlphaZero in JAX☆97Updated 6 months ago
- Portfolio REgret for Confidence SEquences☆20Updated 7 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 5 months ago
- ☆167Updated 3 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆51Updated 2 weeks ago
- σ-GPT: A New Approach to Autoregressive Models☆65Updated 11 months ago
- Evaluating the Mamba architecture on the Othello game☆47Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆103Updated this week
- We integrate discrete diffusion models with neurosymbolic predictors for scalable and calibrated learning and reasoning☆39Updated last month