jax-ml / ml_dtypesLinks
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
☆273Updated 3 weeks ago
Alternatives and similar repositories for ml_dtypes
Users that are interested in ml_dtypes 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.☆112Updated this week
- OpTree: Optimized PyTree Utilities☆186Updated this week
- ☆317Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆400Updated 3 weeks ago
- TorchFix - a linter for PyTorch-using code with autofix support☆143Updated 4 months ago
- High-Performance SGEMM on CUDA devices☆95Updated 5 months ago
- JAX-Toolbox☆311Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- ☆168Updated last year
- A library for unit scaling in PyTorch☆125Updated 6 months ago
- Fast low-bit matmul kernels in Triton☆322Updated this week
- ☆186Updated 2 weeks ago
- PyTorch centric eager mode debugger☆47Updated 6 months ago
- ☆126Updated last month
- extensible collectives library in triton☆86Updated 2 months ago
- ☆219Updated this week
- Named Tensors for Legible Deep Learning in JAX☆181Updated this week
- Implementation of Flash Attention in Jax☆213Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆157Updated 6 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆390Updated this week
- Experiment of using Tangent to autodiff triton☆79Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆510Updated last week
- Collection of kernels written in Triton language☆128Updated 2 months ago
- TritonParse is a tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source code…☆93Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆166Updated this week
- PyTorch per step fault tolerance (actively under development)☆329Updated this week
- ☆52Updated 10 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆556Updated last week
- MLIR-based partitioning system☆91Updated this week
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆94Updated this week