google / array_record
☆72Updated this 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 last month
- ☆84Updated this week
- ☆48Updated 3 months ago
- ☆177Updated this week
- Fast and vectorizable algorithms for searching in a vector of sorted floating point numbers☆119Updated last year
- ☆12Updated 3 years ago
- ☆262Updated this week
- ☆98Updated 4 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆340Updated last week
- A Python library for defining flat or nested dictionary flags.☆30Updated last year
- ☆121Updated this week
- an environment based on XLA for deep learning compiler optimization research.☆23Updated last year
- Pybind11 bindings for the Abseil C++ Common Libraries☆23Updated last month
- ☆104Updated this week
- JAX-Toolbox☆241Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆208Updated this week
- JAX for Graphcore IPU (experimental)☆21Updated 7 months ago
- ☆36Updated 5 months ago
- A tracing JIT compiler for PyTorch☆12Updated 2 years ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆456Updated last week
- LLM training in simple, raw C/CUDA☆86Updated 6 months ago
- extensible collectives library in triton☆61Updated last month
- Implementation of Flash Attention in Jax☆194Updated 8 months ago
- Bazel wrapper around the pybind11 repository☆103Updated 2 weeks ago
- A library for unit scaling in PyTorch☆105Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆296Updated this week
- A simple library for scaling up JAX programs☆125Updated last week
- ☆17Updated 2 weeks ago
- ☆71Updated this week