AI RESEARCH
Parax v0.5: Parametric Modeling in JAX [P]
r/MachineLearning
•
Hi everyone! Just sharing an update on my project Parax, which caters for "parametric modeling" in JAX. Previously, Parax was 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, as well as ensuring the library is entirely opt-in, as opposed to its previous versions which took a framework-like approach.