stockeh / mlx-grokkingLinks
Grokking on modular arithmetic in less than 150 epochs in MLX
☆14Updated 8 months ago
Alternatives and similar repositories for mlx-grokking
Users that are interested in mlx-grokking are comparing it to the libraries listed below
Sorting:
- ☆27Updated last year
- PCCL (Prime Collective Communications Library) implements fault tolerant collective communications over IP☆96Updated last month
- Solidity contracts for the decentralized Prime Network protocol☆23Updated last week
- Modded vLLM to run pipeline parallelism over public networks☆37Updated last month
- train entropix like a champ!☆20Updated 9 months ago
- SIMD quantization kernels☆73Updated last week
- TOPLOC: is a novel method for verifiable inference that enables users to verify that LLM providers are using the correct model configurat…☆34Updated 3 months ago
- Simple Transformer in Jax☆138Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆101Updated 4 months ago
- peer-to-peer compute and intelligence network that enables decentralized AI development at scale☆84Updated this week
- ☆13Updated 8 months ago
- ☆38Updated 11 months ago
- Modular Agentic AI Architecture - NousResearch x Teleport (Flashbots)☆71Updated 6 months ago
- A synthetic story narration dataset to study small audio LMs.☆32Updated last year
- Jax like function transformation engine but micro, microjax☆33Updated 8 months ago
- It's a baby compiler. (Lean btw.)☆16Updated last month
- ☆90Updated this week
- look how they massacred my boy☆63Updated 9 months ago
- ☆13Updated last year
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆70Updated 5 months ago
- Turing machines, Rule 110, and A::B reversal using Claude 3 Opus.☆58Updated last year
- Approximating the joint distribution of language models via MCTS☆21Updated 8 months ago
- train with kittens!☆61Updated 8 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆110Updated last year
- DeMo: Decoupled Momentum Optimization☆189Updated 7 months ago
- ☆19Updated 6 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 8 months ago
- A collection of optimizers for MLX☆36Updated last month
- aesthetic tensor visualiser☆24Updated 2 months ago
- A visual interface for understanding and interpreting Transformers☆77Updated last year