A latent text-to-image diffusion model

Neural Networks: Zero to Hero

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

FaceChain is a deep-learning toolchain for generating your Digital-Twin.

Taming Transformers for High-Resolution Image Synthesis

Efficient Image Captioning code in Torch, runs on GPU

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

Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, B…

Pure Python from-scratch zero-dependency implementation of Bitcoin for educational purposes

Code for the Interspeech 2021 paper "AST: Audio Spectrogram Transformer".

Notebooks and various random fun

Normalizing flows in PyTorch. Current intended use is education not production.

Multi-label Classification with BERT; Fine Grained Sentiment Analysis from AI challenger

VQVAEs, GumbelSoftmaxes and friends

Instruct-tune LLaMA on consumer hardware

Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)

Attention over nodes in Graph Neural Networks using PyTorch (NeurIPS 2019)

⚡️🧪 Fast LLM Tool Calling Experimentation, big and smol

Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.

SImple notebook and dataset to demonstrate classification in TensorFlow

Abstractive and Extractive Text summarization using Transformers.

A pytorch implementation of the paper "Auditing Privacy Defenses in Federated Learning via Generative Gradient Leakage".

PyTorch code for our BMVC 2019 paper "Image Classification with Hierarchical Multigraph Networks"

An InformationGain based Question Answering over knowledge Graph system.

Diverse Image Captioning with Context-Object Split Latent Spaces (NeurIPS 2020)

[Paper] Repository for the paper "On the Relation of Gene Essentiality to Intron Structure: A Computational and Deep Learning Approach", published in Life Science Alliance.