Alternatives and similar repositories for chai:

Users that are interested in chai are comparing it to the libraries listed below

• cloneofsimo / fim-llama-deepspeed
  ☆31Updated last year
• apple / ml-dataset-decomposition
  Official repo of dataset-decomposition paper [NeurIPS 2024]
  ☆13Updated 2 months ago
• cloneofsimo / zeroshampoo
  ☆33Updated 6 months ago
• SmerkyG / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆10Updated 2 weeks ago
• kyegomez / MobileVLM
  Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …
  ☆16Updated last year
• graphcore-research / jax-scalify
  JAX Scalify: end-to-end scaled arithmetics
  ☆16Updated 5 months ago
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆28Updated 3 weeks ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆17Updated 2 weeks ago
• lucidrains / quartic-transformer
  Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)
  ☆47Updated last week
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆20Updated 2 months ago
• brianfitzgerald / jax-mmdit
  Implementation of Diffusion Transformers and Rectified Flow in Jax
  ☆21Updated 8 months ago
• ai-compiler-study / triton-kernels
  Triton kernels for Flux
  ☆20Updated 3 months ago
• recursal / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆45Updated 8 months ago
• kyegomez / USM
  Implementation of Google's USM speech model in Pytorch
  ☆30Updated 2 months ago
• Bartelds / ctc-dro
  Code associated with the paper: CTC-DRO: Robust Optimization for Reducing Language Disparities in Speech Recognition.
  ☆14Updated 3 weeks ago
• chu-tianxiang / QuIP-for-all
  QuIP quantization
  ☆52Updated last year
• facebookresearch / DIG-In
  This library supports evaluating disparities in generated image quality, diversity, and consistency between geographic regions.
  ☆20Updated 9 months ago
• bentherien / mu_learned_optimization
  [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers
  ☆12Updated 2 weeks ago
• SonicCodes / subcloning
  implementation of https://arxiv.org/pdf/2312.09299
  ☆20Updated 8 months ago
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆60Updated 11 months ago
• taehong-moon / ee-diffusion
  Official Code Implementation for 'A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models'
  ☆16Updated 8 months ago
• google-research / last
  A JAX library for building lattice-based speech transducer models
  ☆43Updated 3 months ago
• diggerdu / AudioMamba
  ☆10Updated 10 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 10 months ago
• CLAIRE-Labo / StructuredFFN
  The official code of "Building on Efficient Foundations: Effectively Training LLMs with Structured Feedforward Layers"
  ☆19Updated 8 months ago
• fal-ai-community / NativeSparseAttention
  research impl of Native Sparse Attention (2502.11089)
  ☆53Updated last month
• kyegomez / FastFF
  Zeta implementation of a reusable and plug in and play feedforward from the paper "Exponentially Faster Language Modeling"
  ☆15Updated 4 months ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆55Updated 10 months ago
• davisyoshida / abnormal-floats
  Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)
  ☆18Updated last year
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆48Updated 3 years ago