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 6 months ago
• AI-Hypercomputer / torchprime
  torchprime is a reference model implementation for PyTorch on TPU.
  ☆40Updated last month
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆77Updated 2 months ago
• AI-Hypercomputer / tpu-recipes
  ☆53Updated last week
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆24Updated this week
• mgmalek / efficient_cross_entropy
  ☆121Updated last year
• TobiasNorlund / retro
  Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers
  ☆44Updated last year
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆110Updated 2 weeks ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆86Updated 3 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 3 years ago
• cat-state / tinypar
  ☆20Updated 2 years ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 4 months ago
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆161Updated this week
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆122Updated 9 months ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆54Updated 2 months ago
• google / saxml
  ☆145Updated last week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆326Updated last week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆271Updated last week
• google / praxis
  ☆190Updated 3 weeks ago
• kyo-takano / chinchilla
  A toolkit for scaling law research ⚖
  ☆53Updated 9 months ago
• huggingface / bloom-jax-inference
  ☆66Updated 3 years ago
• samsja / muon_fsdp_2
  Muon fsdp 2
  ☆44Updated 3 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆216Updated this week
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆116Updated 2 months ago
• muellerzr / import-timer
  Pragmatic approach to parsing import profiles for CI's
  ☆12Updated last year
• PythonNut / superbpe
  Official code release for "SuperBPE: Space Travel for Language Models"
  ☆75Updated 2 weeks ago
• epfml / dynamic-sparse-flash-attention
  ☆149Updated 2 years ago
• google-deepmind / asyncdiloco
  ☆47Updated last year
• microsoft / encoder-decoder-slm
  Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…
  ☆32Updated 9 months 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/…
  ☆30Updated 8 months ago