jax-ml/ml_dtypes jax-ml/ml_dtypes

  • Stars
    star
    143
  • Rank 257,007 (Top 6 %)
  • Language
    C++
  • License
    Apache License 2.0
  • Created about 2 years ago
  • Updated 4 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
jax-ml/ml_dtypes
github

jax-ml

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A stand-alone implementation of several NumPy dtype extensions used in machine learning.

More Repositories

1

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python
30,340
star
2

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Python
333
star
3

oryx

Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Python
217
star
4

bayeux

State of the art inference for your bayesian models.
Python
154
star
5

coix

Inference Combinators in JAX
Jupyter Notebook
43
star
6

australis

C++
29
star
7

jax-tpu-embedding

Python
7
star