Alternatives and similar repositories for mlx-grokking:

Users that are interested in mlx-grokking are comparing it to the libraries listed below

• okarthikb / state-space-models
  ☆27Updated 6 months ago
• stockeh / mlx-optimizers
  A collection of optimizers for MLX
  ☆29Updated this week
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 3 months ago
• joey00072 / Attention-as-graph
  alternative way to calculating self attention
  ☆18Updated 8 months ago
• SpellcraftAI / turing
  Turing machines, Rule 110, and A::B reversal using Claude 3 Opus.
  ☆59Updated 8 months ago
• doomslide / autoloom
  ☆20Updated 2 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆130Updated 7 months ago
• xjdr-alt / muzero_sketch
  ☆37Updated 6 months ago
• advait / c4a0
  Alpha-Zero Connect Four NN trained via self play
  ☆13Updated 4 months ago
• samefarrar / entropix_mlx
  Modify Entropy Based Sampling to work with Mac Silicon via MLX
  ☆49Updated 2 months ago
• JD-P / RetroInstruct
  Synthetic data derived by templating, few shot prompting, transformations on public domain corpora, and monte carlo tree search.
  ☆30Updated last month
• 1rgs / token-trekker-rs
  ☆13Updated last year
• nreHieW / r-nn
  Tensor library with autograd using only Rust's standard library
  ☆64Updated 6 months ago
• sfcompute / tinynarrations
  A synthetic story narration dataset to study small audio LMs.
  ☆31Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆45Updated 2 months ago
• xjdr-alt / entropix-trainer
  train entropix like a champ!
  ☆19Updated 3 months ago
• andersonbcdefg / dpo-lora
  direct preference optimization with only 1 model copy :)
  ☆13Updated last year
• teknium1 / RawTransform
  A repository of prompts and Python scripts for intelligent transformation of raw text into diverse formats.
  ☆30Updated last year
• xjdr-alt / llmri
  look how they massacred my boy
  ☆63Updated 3 months ago
• smolorg / smoltropix
  MLX port for xjdr's entropix sampler (mimics jax implementation)
  ☆62Updated 2 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆18Updated last week
• packquickly / schedule_free_optx
  Schedule free optimiser implemented in JAX using Optimistix
  ☆14Updated 8 months ago
• Laz4rz / ziglings
  Extensive introductory writeup on Zig language functionalities
  ☆10Updated 6 months ago
• notarussianteenager / srf-attention
  Simplex Random Feature attention, in PyTorch
  ☆72Updated last year
• leloykun / modded-nanogpt
  NanoGPT (124M) quality in 2.67B tokens
  ☆27Updated this week
• ekzhang / nysrg-notes
  Eric's personal notes from NYSRG https://notes.ekzhang.com/events/nysrg
  ☆14Updated 3 weeks ago
• lumpenspace / FRAG
  Flexible, efficient, and context-aware generation from large unstructured knowledge sources.
  ☆15Updated 8 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆59Updated 6 months ago
• NousResearch / nousflash-agents
  Modular Agentic AI Architecture - NousResearch x Teleport (Flashbots)
  ☆53Updated 3 weeks ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated this week