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 / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆79Updated last month
• AI-Hypercomputer / tpu-recipes
  ☆72Updated last week
• mgmalek / efficient_cross_entropy
  ☆124Updated last year
• google / saxml
  ☆152Updated last month
• google / praxis
  ☆192Updated 3 weeks ago
• cat-state / tinypar
  ☆20Updated 2 years 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
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆113Updated 3 months ago
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆121Updated last month
• microsoft / encoder-decoder-slm
  Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…
  ☆33Updated last year
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆87Updated 3 years ago
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆26Updated last month
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆56Updated 4 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/…
  ☆34Updated 11 months ago
• sholtodouglas / scalingExperiments
  ☆63Updated 3 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 3 years ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 8 months ago
• TobiasNorlund / retro
  Official repo to On the Generalization Ability of Retrieval-Enhanced Transformers
  ☆46Updated last year
• 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
• ltgoslo / gpt-bert
  Official implementation of "GPT or BERT: why not both?"
  ☆61Updated 6 months ago
• kyo-takano / chinchilla
  A toolkit for scaling law research ⚖
  ☆55Updated last year
• HeegyuKim / torch-xla-SPMD
  Pytorch/XLA SPMD Test code in Google TPU
  ☆23Updated last year
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆133Updated 6 months ago
• google / aqt
  ☆345Updated last week
• huggingface / bloom-jax-inference
  ☆66Updated 3 years ago
• google / flaxformer
  ☆367Updated last year