jax-ml / ml_dtypes
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
☆238Updated this week
Alternatives and similar repositories for ml_dtypes:
Users that are interested in ml_dtypes are comparing it to the libraries listed below
- ☆275Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆104Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆363Updated this week
- OpTree: Optimized PyTree Utilities☆163Updated this week
- Named Tensors for Legible Deep Learning in JAX☆157Updated last week
- extensible collectives library in triton☆76Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆219Updated 5 months ago
- Fast low-bit matmul kernels in Triton☆187Updated last week
- Implementation of Flash Attention in Jax☆204Updated 10 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆122Updated last week
- A library for unit scaling in PyTorch☆118Updated last month
- seqax = sequence modeling + JAX☆136Updated 6 months ago
- ☆50Updated 5 months ago
- Experiment of using Tangent to autodiff triton☆74Updated 11 months ago
- ☆170Updated last week
- Orbax provides common checkpointing and persistence utilities for JAX users☆325Updated this week
- JAX-Toolbox☆277Updated this week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆114Updated last year
- ☆106Updated this week
- Applied AI experiments and examples for PyTorch☆211Updated this week
- ☆181Updated 3 weeks ago
- A simple library for scaling up JAX programs☆129Updated 2 months ago
- Cataloging released Triton kernels.☆155Updated last week
- Implementation of a Transformer, but completely in Triton☆251Updated 2 years ago
- Collection of kernels written in Triton language☆90Updated 2 months ago
- ☆20Updated 2 months ago
- ☆154Updated 7 months ago
- A safetensors extension to efficiently store sparse quantized tensors on disk☆64Updated this week
- JMP is a Mixed Precision library for JAX.☆189Updated last month
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆505Updated 2 months ago