google / array_record
☆73Updated last week
Related projects ⓘ
Alternatives and complementary repositories for array_record
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated 2 months ago
- ☆48Updated 3 months ago
- ☆178Updated last week
- ☆12Updated 3 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated 3 weeks ago
- ☆267Updated last week
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆175Updated 2 weeks ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆303Updated this week
- ☆99Updated 4 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆215Updated this week
- A Python library for defining flat or nested dictionary flags.☆30Updated last year
- ☆122Updated this week
- ☆303Updated this week
- extensible collectives library in triton☆71Updated last month
- Implementation of Flash Attention in Jax☆196Updated 8 months ago
- Fast and vectorizable algorithms for searching in a vector of sorted floating point numbers☆120Updated last year
- OpTree: Optimized PyTree Utilities☆152Updated this week
- PyTorch RFCs (experimental)☆130Updated 2 months ago
- A simple library for scaling up JAX programs☆127Updated 2 weeks ago
- A tensor-aware point-to-point communication primitive for machine learning☆249Updated last year
- AutoBound automatically computes upper and lower bounds on functions.☆355Updated 4 months ago
- LLM training in simple, raw C/CUDA☆86Updated 6 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆457Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated 2 weeks ago
- JAX for Graphcore IPU (experimental)☆21Updated 8 months ago
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆81Updated this week
- ☆36Updated 5 months ago
- Data compression in JAX☆56Updated last month
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago