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 last month
Alternatives and similar repositories for zephyr
Users that are interested in zephyr are comparing it to the libraries listed below
Sorting:
- ☆28Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆102Updated last month
- 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.☆126Updated last week
- Schedule free optimiser implemented in JAX using Optimistix☆15Updated last year
- SIMD quantization kernels☆91Updated 2 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆58Updated 3 weeks ago
- nanoGPT using Equinox☆13Updated 2 years ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆227Updated last week
- Code for the Fractured Entangled Representation Hypothesis position paper!☆203Updated 5 months ago
- 🧱 Modula software package☆300Updated 2 months ago
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆104Updated 10 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆62Updated 2 years ago
- An implementation of AlphaZero and MCTS with neural networks for Tetris☆22Updated 7 months ago
- σ-GPT: A New Approach to Autoregressive Models☆68Updated last year
- This repo contains the source code for the paper "Evolution Strategies at Scale: LLM Fine-Tuning Beyond Reinforcement Learning"☆239Updated last week
- Machine Learning library for the emerging Mojo/Python ecosystem☆290Updated last week
- Cost aware hyperparameter tuning algorithm☆172Updated last year
- Tensor library with autograd using only Rust's standard library☆70Updated last year
- Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural Stacks, Queues, Arrays, Lists, Trees, and Latches.☆280Updated last year
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆71Updated 5 months ago
- ☆198Updated 2 months ago
- ☆106Updated 11 months ago
- Attention Kernels for Symmetric Power Transformers☆123Updated last month
- Training AI for Super Smash Bros. Melee☆30Updated 7 months ago
- Gradient descent is cool and all, but what if we could delete it?☆104Updated 2 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 6 months ago
- DeMo: Decoupled Momentum Optimization☆196Updated 11 months ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆69Updated 10 months ago