Alternatives and similar repositories for airio:

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

• google / saxml
  ☆132Updated 2 weeks ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆60Updated this week
• google / t5patches
  T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.
  ☆12Updated 8 months 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…
  ☆103Updated this week
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆51Updated last week
• GoogleCloudPlatform / ml-testing-accelerators
  Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
  ☆64Updated 2 months ago
• google / init2winit
  ☆74Updated last week
• google / uv-metrics
  Composable metric reporters in Python.
  ☆13Updated 8 months ago
• huggingface / gaia
  Hugging Face and Pyserini interoperability
  ☆21Updated last year
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 8 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…
  ☆478Updated 2 weeks ago
• huggingface / bloom-jax-inference
  ☆65Updated 2 years ago
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆114Updated 8 months ago
• google / praxis
  ☆183Updated last week
• jax-ml / jax-tpu-embedding
  ☆14Updated this 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…
  ☆35Updated this week
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆338Updated this week
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆98Updated 3 weeks ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆161Updated this week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆280Updated this week
• huggingface / doc-builder
  The package used to build the documentation of our Hugging Face repos
  ☆104Updated 3 weeks ago
• google / grain
  Library for reading and processing ML training data.
  ☆385Updated this week
• hauntsaninja / boostedblob
  Command line tool and async library to perform basic file operations on local paths, Google Cloud Storage paths and Azure Blob Storage pa…
  ☆25Updated last month
• CarperAI / treasure_trove
  ☆22Updated last year
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆168Updated 7 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆252Updated 2 weeks ago
• google / merchant_center_repor_builder
  ☆11Updated 7 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆107Updated 3 weeks 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…
  ☆274Updated this week
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆86Updated this week