Giters
jax-ml

jax-ml

Geek Repo

miscellaneous libraries and projects relating to JAX

Home Page:github.com/google/jax

Github PK Tool:Github PK Tool

jax-ml's repositories

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Language:PythonLicense:Apache-2.0Stargazers:311Issues:21Issues:54

oryx

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

Language:PythonLicense:Apache-2.0Stargazers:202Issues:9Issues:18

ml_dtypes

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

Language:C++License:Apache-2.0Stargazers:143Issues:9Issues:36

bayeux

State of the art inference for your bayesian models.

Language:PythonLicense:Apache-2.0Stargazers:106Issues:3Issues:11

coix

Inference Combinators in JAX

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:41Issues:7Issues:9

australis

Language:C++License:Apache-2.0Stargazers:29Issues:3Issues:0

jax-tpu-embedding

Language:PythonLicense:Apache-2.0Stargazers:6Issues:1Issues:0

.allstar

Stargazers:0Issues:1Issues:0

.github

Stargazers:0Issues:1Issues:0