Stars
2
results
for sponsorable starred repositories
Clear filter
Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/