Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
APACHE-2.0 License
A language for mental models
Gaussian Processes for Experimental Sciences
Train very large language models in Jax.
Learn about the Neumorphic engineering process of creating large-scale integration (VLSI) systems...
UNKNOWN
A miscellaneous set of helper functions, custom distributions, and other utilities that I find us...
some tools for gaussian linear dynamical systems
Decorator for PyMC3
v objective diffusion inference code for JAX.
Bayesian Inference in JAX
Concise deep learning for JAX
Simulation-based inference in JAX
Find better generation parameters for your LLM
Multiple dispatch over abstract array types in JAX.