Pytree
4 projects
Showing 4 of 4 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
Forks207
Last commit24 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
Stars948
Forks70
Last commit2 days ago
flowjaxPython
A JAX library for distributions, bijections, and normalizing flows implemented as Equinox modules.
#probabilistic-modeling#jax#normalizing-flows
Stars238
Forks24
Last commit1 month ago
LoraxPython
A JAX transform that implements LoRA (Low-Rank Adaptation) for efficient fine-tuning of large models with minimal memory overhead.
#parameter-efficient-training#jax#deep-learning
Stars143
Forks6
Last commit2 years 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.