Alternatives and similar repositories for kithara

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

• AI-Hypercomputer / ray-tpu
  ☆15Updated 8 months ago
• AI-Hypercomputer / torchprime
  torchprime is a reference model implementation for PyTorch on TPU.
  ☆44Updated last month
• AI-Hypercomputer / tpu-recipes
  ☆72Updated last week
• mgmalek / efficient_cross_entropy
  ☆124Updated last year
• google / praxis
  ☆192Updated this week
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆79Updated last month
• google / saxml
  ☆152Updated last month
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆26Updated last month
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆87Updated 3 years ago
• kyo-takano / chinchilla
  A toolkit for scaling law research ⚖
  ☆55Updated last year
• cat-state / tinypar
  ☆20Updated 2 years ago
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆121Updated last month
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 3 years ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆389Updated last week
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆135Updated 2 weeks ago
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆228Updated this week
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆113Updated 3 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆133Updated 6 months ago
• TobiasNorlund / retro
  Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers
  ☆46Updated last year
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆56Updated 4 months ago
• AI-Hypercomputer / xpk
  xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…
  ☆162Updated this week
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 8 months ago
• CLAIRE-Labo / RAT
  Official code for the NeurIPS25 paper "RAT: Bridging RNN Efficiencyand Attention Accuracy in Language Modeling" (https://arxiv.org/abs/25…
  ☆23Updated last month
• samsja / muon_fsdp_2
  Muon fsdp 2
  ☆52Updated 6 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆279Updated 2 months ago
• TurkuNLP / Megatron-DeepSpeed
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆20Updated 2 years ago
• erfanzar / jax-flash-attn2
  A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…
  ☆34Updated 11 months ago
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆547Updated 3 weeks ago
• ZurichNLP / mbr
  Minimum Bayes Risk Decoding for Hugging Face Transformers
  ☆60Updated last year
• NVIDIA-NeMo / Emerging-Optimizers
  ☆147Updated this week