Code

Open-source releases accompanying the papers above. More at github.com/dbaranchuk.

yandex-research/switti (CVPR 2025) Switti: Designing Scale-Wise Transformers for Text-to-Image Synthesis Jupyter Notebook 205 yandex-research/swd [ICLR'2026] Scale-wise Distillation of Diffusion Models Python 124 yandex-research/ar-classifier [ECCV 2026] Official code of "Revisiting Autoregressive Models for Generative Image Classification" Python 14 yandex-research/invertible-cd [NeurIPS'2024] Invertible Consistency Distillation for Text-Guided Image Editing in Around 7 Steps Python 101 yandex-research/vqdm Official repo for VQDM, a paper on compressing text-to-image diffusion models with vector quantization Python 33 yandex-research/adaptive-diffusion [CVPR'2024] Adaptive Teacher-Student Collaboration for Text-Conditional Diffusion Models Python 33 yandex-research/tab-ddpm [ICML 2023] The official implementation of the paper "TabDDPM: Modelling Tabular Data with Diffusion Models" Python 563 yandex-research/ddpm-segmentation Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022) Python 718 bigscience-workshop/petals Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading Python 10.5k harsha-simhadri/big-ann-benchmarks Framework for evaluating ANNS algorithms on billion scale datasets. Jupyter Notebook 448 dbaranchuk/ivf-hnsw Code for ECCV2018 paper: Revisiting the Inverted Indices for Billion-Scale Approximate Nearest Neighbors C++ 218 dbaranchuk/learning-to-route Code for ICML2019 paper: Learning to Route in Similarity Graphs Jupyter Notebook 62 dbaranchuk/nns-meets-deep-rl Code for paper: Towards Similarity Graphs Constructed by Deep Reinforcement Learning C 21 ratschlab/GP-VAE TensorFlow implementation for the GP-VAE model described in https://arxiv.org/abs/1907.04155 Python 144 dbaranchuk/memory-efficient-maml Memory efficient MAML using gradient checkpointing Jupyter Notebook 86 yandex-research/distill-nf Code for the paper: Distilling the Knowledge from Conditional Normalizing Flows Jupyter Notebook 9 yandex-research/learnable-init Code for the paper: Discovering Weight Initializers with Meta-Learning Jupyter Notebook 5