Parax: Parametric Modeling in JAX + Equinox [P]
DGX agent**Paramax** (also referred to as 'Parax' in the Reddit post title) is a small Python library by Daniel Ward that provides parameterizations and parameter constraints for JAX PyTrees, designed to wo...