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.
  ☆28Updated last week
• google / saxml
  ☆141Updated 3 weeks ago
• AI-Hypercomputer / tpu-recipes
  ☆34Updated last week
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆62Updated 2 months ago
• AI-Hypercomputer / cloud-accelerator-diagnostics
  ☆21Updated last week
• gclouduniverse / reproducibility-deprecated
  ☆16Updated 3 months ago
• AI-Hypercomputer / gpu-recipes
  Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
  ☆73Updated last week
• GoogleCloudPlatform / ml-auto-solutions
  A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across differe…
  ☆48Updated 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…
  ☆349Updated 2 weeks ago
• GoogleCloudPlatform / ml-testing-accelerators
  Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
  ☆64Updated 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…
  ☆125Updated this week
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆113Updated 5 months ago
• google / praxis
  ☆186Updated 3 weeks ago
• pytorch-labs / monarch
  PyTorch Single Controller
  ☆231Updated this week
• rwitten / HighPerfLLMs2024
  ☆504Updated 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…
  ☆510Updated last week
• AI-Hypercomputer / ml-goodput-measurement
  ☆17Updated last week
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆118Updated last year
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• jax-ml / jax-llm-examples
  ☆126Updated last month
• google / tunix
  A JAX-native LLM Post-Training Library
  ☆58Updated this week
• google / array_record
  ☆87Updated last week
• pytorch / torchft
  PyTorch per step fault tolerance (actively under development)
  ☆329Updated this week
• google / aqt
  ☆318Updated this week
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆393Updated this week
• huggingface / bloom-jax-inference
  ☆67Updated 2 years ago
• danielvegamyhre / ml-perf-reading-group
  My annotated papers and meeting recordings for the EleutherAI ML Performance research paper reading group
  ☆13Updated last month
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆596Updated this week
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆179Updated 2 weeks ago
• sholtodouglas / multihost_dataloading
  Experimenting with how best to do multi-host dataloading
  ☆10Updated 2 years ago