Explore @jax-ml Open Source projects

@jax-ml

jax-ml

Stars
300
Global Org. Rank 32,322 (Top 11 %)
Followers 64
Registered almost 5 years ago
Most used languages

Python
62.5 %

C++
25.0 %

Jupyter Notebook
12.5 %

jax

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

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

oryx

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

bayeux

State of the art inference for your bayesian models.

ml_dtypes

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

coix

Inference Combinators in JAX

Jupyter Notebook

australis

jax-tpu-embedding