Alternatives and similar repositories for kithara

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

• AI-Hypercomputer / torchprime
  torchprime is a reference model implementation for PyTorch on TPU.
  ☆39Updated last week
• AI-Hypercomputer / ray-tpu
  ☆15Updated 5 months ago
• google / saxml
  ☆145Updated 2 weeks ago
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆76Updated last month
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆23Updated last week
• AI-Hypercomputer / tpu-recipes
  ☆49Updated last week
• 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…
  ☆147Updated this week
• AI-Hypercomputer / JetStream
  JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…
  ☆385Updated 4 months ago
• google / praxis
  ☆190Updated this week
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆539Updated last month
• AI-Hypercomputer / gpu-recipes
  Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
  ☆93Updated this week
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆123Updated this week
• google / aqt
  ☆335Updated last month
• rwitten / HighPerfLLMs2024
  ☆542Updated last year
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆121Updated 9 months ago
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• GoogleCloudPlatform / ml-testing-accelerators
  Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
  ☆65Updated 4 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆436Updated last week
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆311Updated 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…
  ☆270Updated 3 months ago
• AI-Hypercomputer / ml-goodput-measurement
  ☆24Updated last month
• google / qwix
  a Jax quantization library
  ☆52Updated this week
• mgmalek / efficient_cross_entropy
  ☆121Updated last year
• huggingface / kernels
  Load compute kernels from the Hub
  ☆308Updated this week
• marin-community / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆671Updated this week
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 3 months ago
• AI-Hypercomputer / maxdiffusion
  ☆269Updated this week
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆215Updated this week
• meta-pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆395Updated last week
• gclouduniverse / reproducibility-deprecated
  ☆16Updated 7 months ago