AI-Hypercomputer / ml-goodput-measurement
☆10Updated last month
Related projects ⓘ
Alternatives and complementary repositories for ml-goodput-measurement
- T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.☆12Updated 5 months ago
- ☆121Updated this week
- ☆26Updated last year
- This repo contains the code associated to the paper: "Constrained Causal Bayesian Optimization" by Aglietti Virginia, Alan Malek, Ira Kt…☆13Updated 4 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆40Updated this week
- struct2tensor is a library for parsing and manipulating structured data inside of tensorflow.☆34Updated last month
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆80Updated this week
- Source-to-Source Debuggable Derivatives in Pure Python☆14Updated 9 months ago
- Code for paper: "Privately generating tabular data using language models".☆14Updated last year
- Composable metric reporters in Python.☆12Updated 5 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆42Updated 5 months ago
- With Vertex AI NAS, you can search for optimal neural architectures in terms of accuracy, latency, memory, a combination of these, or a c…☆25Updated last year
- Libraries for efficient and scalable group-structured dataset pipelines.☆23Updated last month
- ☆35Updated 7 months ago
- Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …☆14Updated 8 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated last month
- ☆34Updated last year
- Modular, scalable codebase to train ML models☆42Updated this week
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64Updated 2 months ago
- ☆13Updated 4 months ago
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Updated last year
- Machine Learning eXperiment Utilities☆45Updated 5 months ago
- ☆25Updated last month
- Prototype routines for GPU quantization written using PyTorch.☆19Updated this week
- Randomized Positional Encodings Boost Length Generalization of Transformers☆79Updated 7 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆52Updated last month
- Scalable and Performant Data Loading☆60Updated this week
- This is the official PyTorch repo for "UNIREX: A Unified Learning Framework for Language Model Rationale Extraction" (ICML 2022).☆23Updated last year
- ☆25Updated 11 months ago