google / array_record
☆81Updated last week
Alternatives and similar repositories for array_record:
Users that are interested in array_record are comparing it to the libraries listed below
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆109Updated last month
- ☆51Updated 7 months ago
- ☆184Updated last month
- OpTree: Optimized PyTree Utilities☆173Updated this week
- Library for reading and processing ML training data.☆409Updated last week
- ☆94Updated this week
- ☆112Updated last month
- Orbax provides common checkpointing and persistence utilities for JAX users☆351Updated this week
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- ☆290Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆387Updated 2 weeks ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆255Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆484Updated 2 weeks ago
- A simple library for scaling up JAX programs☆134Updated 4 months ago
- ☆101Updated 9 months ago
- ☆137Updated this week
- ☆20Updated 3 weeks ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆53Updated 3 weeks ago
- ☆13Updated 3 years ago
- Implementation of Flash Attention in Jax☆206Updated last year
- Riegeli/records is a file format for storing a sequence of string records, typically serialized protocol buffers.☆430Updated last week
- JAX Synergistic Memory Inspector☆171Updated 8 months ago
- Stores documents and resources used by the OpenXLA developer community☆117Updated 7 months ago
- Starlark implementation of bazel rules for CUDA.☆102Updated this week
- JAX-Toolbox☆292Updated this week
- ☆21Updated 3 weeks ago
- CLU lets you write beautiful training loops in JAX.☆335Updated 3 weeks ago
- A functional training loops library for JAX☆86Updated last year
- ☆15Updated this week