scikit-learn

pynear 2.3 is a fast KNN library with SIMD acceleration and drop-in compatibility with scikit-learn, offering exact and approximate cosine indices and outperforming Faiss on binary descriptors in some contexts.

bde is a Python package that brings sampling-based Bayesian Deep Learning to practitioners via the MILE method, combining JAX's speed with scikit-learn's API for tabular supervised learning tasks.