Pytree
3 projects
Showing 3 of 3 projects
EquinoxPython
A JAX library for neural networks and scientific computing with PyTorch-like syntax and full ecosystem compatibility.
#scientific-computing#gradient-based-optimization#jax
Stars2.9k
Forks195
Last commit11 days ago
ChexPython
A library of utilities for writing and testing reliable JAX code, including assertions, debugging tools, and test variants.
#code-reliability#jax#python-library
Stars941
Forks68
Last commit1 month ago
flowjaxPython
A JAX library for distributions, bijections, and normalizing flows implemented as Equinox modules.
#probabilistic-modeling#jax#normalizing-flows
Stars227
Forks22
Last commit3 months ago
Related Tags
Community-curated · Updated weekly · 100% open source
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.