Parax: Parametric Modeling in JAX + Equinox [P]
Hi everyone! Just wanted to share my Python project Parax – an add-on on top of the Equinox library catering for parameter-first modeling in JAX. For our scientific applications, we found that we often needed to attach metadata to our parameter objects…