Tensors and Dynamic neural networks in Python with strong GPU acceleration

🚀 Generate GitHub profile README easily with the latest add-ons like visitors count, GitHub stats, etc using minimal UI.

DALL·E Mini - Generate images from a text prompt

Refine high-quality datasets and visual AI models

Supercharge Your Model Training

Transformer: PyTorch Implementation of "Attention Is All You Need"

Search over large image datasets with natural language and computer vision

C++ implementations of well-known (and some rare) algorithms, while following good software development practices

User-friendly secure computation engine based on secure multi-party computation

Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis Implementation

Red-Teaming Language Models with DSPy

Python package for extracting representations from state-of-the-art computer vision models

PyTorch implementation of VQ-VAE-2 from "Generating Diverse High-Fidelity Images with VQ-VAE-2"

The Official Repository for "Bring Your Own Data! Self-Supervised Evaluation for Large Language Models"

PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)

Gemstones: A Model Suite for Multi-Faceted Scaling Laws (NeurIPS 2025)

A multi-label text classifier to predict the subject areas of arXiv papers from their abstract bodies.

Repository for paper "Non-intrusive speech intelligibility prediction from discrete latent representations"

Probabilistic programming with HuggingFace language models. Updated with additional examples

PyTorch implementation of "Denoising Diffusion Probabilistic Models"

Repository to store my learning experiences with JAX

The 8-puzzle game and also lets the user get it solved by the (A* heuristics based) A.I