[P] eqx-learn: Classical machine learning using JAX and Equinox
This JAX-native ML library could change how researchers build classical models...
Developer Gary Callen has released eqx-learn, a new library bringing classical machine learning models to the JAX/Equinox ecosystem. The library implements a scikit-learn-style API but with JAX-native models that are Equinox Modules. It currently supports PCA and Gaussian Process Regression, with models advertising capabilities through specific methods while a universal "fit" function handles training. The decoupled design aligns with JAX's functional style and aims to fill a gap in the ecosystem for straightforward classical ML tools.
Why It Matters
Researchers using JAX for modern deep learning can now handle classical ML tasks within the same framework, streamlining workflows.