jax-ml / ml_dtypesLinks
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
☆301Updated 2 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:
- ☆335Updated last month
- jax-triton contains integrations between JAX and OpenAI Triton☆428Updated last week
- TorchFix - a linter for PyTorch-using code with autofix support☆149Updated 2 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated last month
- OpTree: Optimized PyTree Utilities☆196Updated last week
- ☆190Updated 3 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆389Updated last week
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆102Updated this week
- Experiment of using Tangent to autodiff triton☆80Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆223Updated last year
- High-Performance SGEMM on CUDA devices☆107Updated 9 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated last month
- ☆178Updated last year
- ☆77Updated last week
- JAX-Toolbox☆355Updated this week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆145Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAX☆210Updated last week
- extensible collectives library in triton☆89Updated 6 months ago
- Implementation of Flash Attention in Jax☆219Updated last year
- ☆240Updated 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…☆180Updated last month
- The CUDA target for Numba☆200Updated last week
- ☆21Updated 7 months ago
- TORCH_LOGS parser for PT2☆62Updated last month
- PyTorch RFCs (experimental)☆135Updated 4 months ago
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- A library for unit scaling in PyTorch☆130Updated 3 months ago
- ☆28Updated 9 months ago
- ☆52Updated last year
- Tokamax: A GPU and TPU kernel library.☆94Updated this week