annax. annax is an ANN library based on JAX that provides fast and memory-efficient approximate nearest neighbor search for high-dimensional data. It includes a wide range of indexing structures and distance metrics for different use cases. With its easy-to-use API and GPU acceleration, Annax is a powerful tool for large-scale ML applications.

github.com/atksh/annax

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.