Paramax is a JAX package for applying custom parameterizations and constraints to PyTree components...

Tokens:8,602
Snippets:66
Trust Score:6.6
License:MIT
Update:2 months ago
Tokens:
Raw