Machine learning algorithms implements with jax for machine learning in production in large scale dataset.
☆24Oct 1, 2026Updated last week
Alternatives and similar repositories for jax-sklearn
Users that are interested in jax-sklearn are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated last month
- Stochastic Gradient MCMC for Jax☆20May 19, 2025Updated last year
- Code for reproducing the paper "Neural Networks Fail to Learn Periodic Functions and How to Fix It" as part of the ML Reproducibility Cha…