bijax

Bayesian Inference in JAX

Downloads
28
Stars
7
Committers
2
View Code on GitHub View on X
Ecosystems: Python

Commit Statistics

Past Year

All Time

Total Commits
63
Total Committers
2
Avg. Commits Per Committer
31.5
Bot Commits
0

Issue Statistics

Past Year

All Time

Total Pull Requests
0
4
Merged Pull Requests
0
2
Total Issues
0
8
Time to Close Issues
N/A
17 days
Package Rankings
Top 27.15% on Pypi.org
Related Projects
ninjax

General Modules for JAX

23 May 2022 57
attax

Attax: adversarial attacks using JAX

27 Oct 2019 3
conjugate

Bayesian Conjugate Models in Python

23 Jun 2023 23
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
sbijax

Simulation-based inference in JAX

12 Oct 2022 12
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
gpax

Gaussian Processes for Experimental Sciences

28 Oct 2021 205
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
pybasicbayes

Basic utilities for Bayesian inference

04 Aug 2012 153
openlogprobs

Extract full next-token probabilities via language model APIs

22 Nov 2023 160
probmods

An interface to facilitate rapid development and deployment of probabilistic models

04 Jul 2021 4
numpyro-ext

A miscellaneous set of helper functions, custom distributions, and other utilities that I find us...

26 Jul 2022 18
mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.

22 Jan 2020 324
jaxnet

Concise deep learning for JAX

11 Aug 2019 184
mlp-gpt-jax

A GPT, made only of MLPs, in Jax

21 May 2021 55