Alternatives and similar repositories for awesome-jax-flax-llms

Users that are interested in awesome-jax-flax-llms are comparing it to the libraries listed below

• okarthikb / state-space-models
  ☆28Updated last year
• conglu1997 / ACD
  Automated Capability Discovery via Foundation Model Self-Exploration
  ☆64Updated 7 months ago
• rimads / avey-dpa
  Code for the paper Don't Pay Attention
  ☆49Updated this week
• kubernetes-bad / reward-composer
  Lego for GRPO
  ☆29Updated 4 months ago
• bradhilton / temporal-clue
  Clue inspired puzzles for testing LLM deduction abilities
  ☆43Updated 6 months ago
• OpenPipe / deductive-reasoning
  Train your own SOTA deductive reasoning model
  ☆106Updated 6 months ago
• upiterbarg / diff_history
  [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)
  ☆20Updated last year
• dhkts1 / sequentialStory
  story based implementation for sequential thinking
  ☆14Updated 2 weeks ago
• reka-ai / rekaquant
  ☆62Updated 2 months ago
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆72Updated 5 months ago
• Oxen-AI / GRPO-With-Cargo-Feedback
  This repository has code for fine-tuning LLMs with GRPO specifically for Rust Programming using cargo as feedback
  ☆104Updated 6 months ago
• idoh / fast_mamba.np
  A pure and fast NumPy implementation of Mamba with cache support.
  ☆17Updated last year
• PrimeIntellect-ai / prime-cli
  The Prime Intellect CLI provides a powerful command-line interface for managing GPU resources across various providers
  ☆91Updated this week
• haizelabs / spoken
  a single interface around speech-to-speech foundation models
  ☆25Updated 3 months ago
• SakanaAI / ALE-Bench
  The official repository of ALE-Bench
  ☆114Updated this week
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆33Updated last year
• andrew-silva / mlx-rlhf
  An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.
  ☆32Updated last year
• MagellaX / StreamAttn
  A high-performance attention mechanism that computes softmax normalization in a single streaming pass using running accumulators (online …
  ☆26Updated 3 weeks ago
• hundredblocks / large-model-parallelism
  Functional local implementations of main model parallelism approaches
  ☆96Updated 2 years ago
• valine / training-hot-swap
  Pytorch script hot swap: Change code without unloading your LLM from VRAM
  ☆125Updated 5 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated last year
• PrimeIntellect-ai / prime-environments
  Training-Ready RL Environments + Evals
  ☆111Updated this week
• Agora-Lab-AI / OmegaViT
  OmegaViT (ΩViT) is a cutting-edge vision transformer architecture that combines multi-query attention, rotary embeddings, state space mod…
  ☆14Updated this week
• brendanhogan / picoDeepResearch
  ☆68Updated 4 months ago
• jxmorris12 / embzip
  lossily compress representation vectors using product quantization
  ☆59Updated 5 months ago
• kuprel / minbpe-pytorch
  Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization, with PyTorch/CUDA
  ☆40Updated last year
• cgftinc / benchmax
  Framework-Agnostic RL Environments for LLM Fine-Tuning
  ☆36Updated last week
• Antlera / nanoGPT-moe
  Enable moe for nanogpt.
  ☆34Updated last year
• guidance-ai / llgtrt
  TensorRT-LLM server with Structured Outputs (JSON) built with Rust
  ☆58Updated 5 months ago
• winfunc / asterisk-mcp-server
  Asterisk Model Context Protocol (MCP) server.
  ☆25Updated 6 months ago