Alternatives and similar repositories for ml-goodput-measurement

Users that are interested in ml-goodput-measurement are comparing it to the libraries listed below

• 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…
  ☆154Updated this week
• google / saxml
  ☆148Updated last month
• google / qwix
  a Jax quantization library
  ☆71Updated this week
• AI-Hypercomputer / tpu-recipes
  ☆54Updated this week
• AI-Hypercomputer / gpu-recipes
  Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
  ☆103Updated this week
• GoogleCloudPlatform / ml-testing-accelerators
  Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
  ☆65Updated 5 months ago
• 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…
  ☆392Updated 5 months ago
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆78Updated 2 months ago
• 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…
  ☆55Updated this week
• google / praxis
  ☆190Updated 2 weeks ago
• google / array_record
  ☆126Updated 3 weeks ago
• AI-Hypercomputer / maxdiffusion
  ☆283Updated last week
• AI-Hypercomputer / kithara
  ☆16Updated 6 months ago
• jax-ml / jax-tpu-embedding
  ☆25Updated this week
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• GoogleCloudPlatform / slurm-gcp
  ☆57Updated 2 weeks ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆176Updated this week
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆122Updated this week
• google / struct2tensor
  struct2tensor is a library for parsing and manipulating structured data inside of tensorflow.
  ☆34Updated 2 months ago
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆144Updated this week
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆123Updated 10 months ago
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆77Updated last week
• google / aqt
  ☆337Updated 2 weeks ago
• AI-Hypercomputer / ray-tpu
  ☆15Updated 6 months ago
• google / t5patches
  T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.
  ☆12Updated last year
• AI-Hypercomputer / torchprime
  torchprime is a reference model implementation for PyTorch on TPU.
  ☆41Updated last month
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆178Updated this week
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated 2 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 9 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆349Updated this week