google / array_recordLinks
☆87Updated last week
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.☆114Updated this week
- ☆186Updated 3 weeks ago
- ☆52Updated 10 months ago
- ☆141Updated 3 weeks ago
- ☆21Updated last week
- ☆318Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆276Updated 3 weeks ago
- ☆104Updated last year
- OpTree: Optimized PyTree Utilities☆186Updated last week
- Library for reading and processing ML training data.☆463Updated this week
- ☆96Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆400Updated 3 weeks ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆393Updated this week
- ☆13Updated 4 years ago
- A simple library for scaling up JAX programs☆139Updated 7 months ago
- ☆114Updated 3 weeks ago
- JAX-Toolbox☆311Updated this week
- JMP is a Mixed Precision library for JAX.☆203Updated 4 months ago
- Implementation of Flash Attention in Jax☆213Updated last year
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- seqax = sequence modeling + JAX☆159Updated last week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆510Updated last week
- MLIR-based partitioning system☆97Updated this week
- ☆18Updated this week
- A functional training loops library for JAX☆88Updated last year
- Named Tensors for Legible Deep Learning in JAX☆181Updated this week
- JAX Synergistic Memory Inspector☆174Updated 11 months ago
- ☆188Updated last week
- ☆270Updated 11 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago