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
• google / saxml
  ☆146Updated 2 weeks ago
• AI-Hypercomputer / ray-tpu
  ☆15Updated 4 months ago
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆23Updated last week
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆73Updated 3 weeks 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…
  ☆143Updated last week
• AI-Hypercomputer / tpu-recipes
  ☆46Updated 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…
  ☆379Updated 3 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…
  ☆538Updated last month
• google / praxis
  ☆189Updated last 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…
  ☆392Updated this week
• AI-Hypercomputer / gpu-recipes
  Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
  ☆88Updated last week
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆414Updated last week
• gclouduniverse / reproducibility-deprecated
  ☆16Updated 6 months ago
• rwitten / HighPerfLLMs2024
  ☆538Updated last year
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆268Updated 2 months ago
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• google / aqt
  ☆331Updated 3 weeks ago
• AI-Hypercomputer / ml-goodput-measurement
  ☆23Updated 3 weeks ago
• GoogleCloudPlatform / ml-testing-accelerators
  Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
  ☆65Updated 3 months ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆120Updated 8 months ago
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 weeks ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆304Updated 2 weeks ago
• sholtodouglas / multihost_dataloading
  Experimenting with how best to do multi-host dataloading
  ☆10Updated 2 years ago
• meta-pytorch / monarch
  PyTorch Single Controller
  ☆425Updated this week
• mgmalek / efficient_cross_entropy
  ☆122Updated last year
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆426Updated last month
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆181Updated 2 weeks ago
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆87Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆293Updated this week