google / array_recordLinks
☆85Updated 2 months ago
Alternatives and similar repositories for array_record
Users that are interested in array_record are comparing it to the libraries listed below
Sorting:
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆110Updated 3 weeks ago
- ☆186Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆268Updated last week
- ☆95Updated this week
- ☆52Updated 9 months ago
- ☆138Updated 2 weeks ago
- A simple library for scaling up JAX programs☆137Updated 7 months ago
- JMP is a Mixed Precision library for JAX.☆199Updated 4 months ago
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- ☆310Updated 2 weeks ago
- ☆114Updated this week
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Updated 8 months ago
- Implementation of Flash Attention in Jax☆212Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆499Updated last week
- ☆104Updated 11 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆384Updated this week
- ☆20Updated 2 weeks ago
- OpTree: Optimized PyTree Utilities☆185Updated this week
- Library for reading and processing ML training data.☆447Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆395Updated this week
- ☆13Updated 4 years ago
- ☆18Updated this week
- JAX Synergistic Memory Inspector☆173Updated 10 months ago
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64Updated last week
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Updated last year
- MLIR-based partitioning system☆86Updated this week
- PyTorch RFCs (experimental)☆132Updated last week
- JAX-Toolbox☆308Updated this week
- A library for unit scaling in PyTorch☆125Updated 6 months ago
- Fast and vectorizable algorithms for searching in a vector of sorted floating point numbers☆140Updated 5 months ago