Hi everyone! Just sharing an update on my project Parax, which caters for "parametric modeling" in JAX. Previously, Parax was more focused on scientific applications, however I've since generalized it to be a tool useful for any type of JAX work. It now has a strong focus on a clean, extandable API,
