A high performance implementation of HDBSCAN clustering.

Python implementations of the k-modes and k-prototypes clustering algorithms, for clustering categorical data

This is the official code for MobileSAM project that makes SAM lightweight for mobile applications and beyond!

Neural Network primitives with multiple backends

Efficient Deep Learning Systems course materials (HSE, YSDA)

A collection of various deep learning architectures, models, and tips

Metric learning and retrieval pipelines, models and zoo.

A General-purpose Task-parallel Programming System using Modern C++

PyTorch native quantization and sparsity for training and inference

Simple, safe way to store and distribute tensors