jax-ml

miscellaneous libraries and projects relating to JAX

Pinned Repositories

.allstar
0 1 00
.github
0 1 00
australis
Language:C++29 3 04
bayeux
State of the art inference for your bayesian models.
Language:Python105 3 104
coix
Inference Combinators in JAX
Language:Jupyter Notebook40 7 92
jax-tpu-embedding
Language:Python6 1 01
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
Language:Python308 23 5430
ml_dtypes
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Language:C++132 9 3419
oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Language:Python197 9 189

jax-ml's Repositories

jax-ml/jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
Language:Python308 23 5430
jax-ml/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Language:Python197 9 189
jax-ml/ml_dtypes
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Language:C++132 9 3419
jax-ml/bayeux
State of the art inference for your bayesian models.
Language:Python105 3 104
jax-ml/coix
Inference Combinators in JAX
Language:Jupyter Notebook40 7 92
jax-ml/australis
Language:C++29 3 04
jax-ml/jax-tpu-embedding
Language:Python6 1 01
jax-ml/.allstar
0 1 00
jax-ml/.github
0 1 00