6 projects
dinox
Derivative Informed Neural Operators in JAX and Equinox
BayesFlux
Bayesian Fast Linear algebra sUbspace eXtraction in JAX
randlax
Randomized Linear Algebra routines for JAX
flowjax-py39
Easy to use distributions, bijections and normalizing flows in JAX, forked for Python 3.9 compatability
paramax-py39
Parameterizations and parameter constraints for JAX PyTrees, forked for Python 3.9 compatability.
semilattices
This package implements a lightweight library for semilattices in python.
