mzguntalan / zephyrLinks
Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulation
☆35Updated 3 months ago
Alternatives and similar repositories for zephyr
Users that are interested in zephyr are comparing it to the libraries listed below
Sorting:
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆61Updated last week
- A pure and fast NumPy implementation of Mamba with cache support.☆17Updated last year
- A reinforcement learning codebase focusing on the emergence of cooperation and alignment in multi-agent AI systems.☆134Updated last week
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆106Updated last month
- ☆28Updated last year
- nanoGPT using Equinox☆14Updated 2 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆64Updated this week
- 🧱 Modula software package☆316Updated 4 months ago
- Tensor library with autograd using only Rust's standard library☆70Updated last year
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆105Updated 11 months ago
- Reinforcement learning with Equinox☆19Updated 9 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆237Updated last month
- Cost aware hyperparameter tuning algorithm☆176Updated last year
- This repo contains the source code for the paper "Evolution Strategies at Scale: LLM Fine-Tuning Beyond Reinforcement Learning"☆277Updated last month
- Schedule free optimiser implemented in JAX using Optimistix☆15Updated last year
- Code for the Fractured Entangled Representation Hypothesis position paper!☆217Updated last month
- Grokking on modular arithmetic in less than 150 epochs in MLX☆14Updated last year
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- SIMD quantization kernels☆93Updated 3 months ago
- An implementation of Tiny Recursive Models (TRM)☆80Updated this week
- ☆44Updated last month
- A compositional diagramming and animation library as an eDSL in Python☆217Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆62Updated 2 years ago
- A flexible and extensible reinforcement learning library for Python, designed for both beginners and researchers.☆18Updated last year
- ☆211Updated 4 months ago
- An implementation of AlphaZero and MCTS with neural networks for Tetris☆22Updated 9 months ago
- Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural Stacks, Queues, Arrays, Lists, Trees, and Latches.☆284Updated last year
- Portfolio REgret for Confidence SEquences☆20Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆72Updated last year
- Neural Networks for JAX☆84Updated last year