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.