AI RESEARCH
Parax v0.7: Parametric Modeling in JAX [P]
r/MachineLearning
•
Hi everyone! Parax is a library for "Parametric modeling" in JAX, attempting to bridge the approach between pure JAX PyTrees, and object-orientated modeling approaches (e.g. using Equinox ). v0.7 has been released, featuring a polished API as well as some detailed examples in the documentation.