jax-ml/bayeux jax-ml/bayeux

  • Stars
    star
    154
  • Rank 242,095 (Top 5 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created about 1 year ago
  • Updated about 2 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
jax-ml/bayeux
github

jax-ml

Reviews

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

Repository Details

State of the art inference for your bayesian models.

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

ml_dtypes

A stand-alone implementation of several NumPy dtype extensions used in machine learning.
C++
143
star
5

coix

Inference Combinators in JAX
Jupyter Notebook
43
star
6

australis

C++
29
star
7

jax-tpu-embedding

Python
7
star