PierreHao

3D-Machine-Learning

A resource repository for 3D machine learning

3DInfomax

Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.

APE

Aligning and Prompting Everything All at Once for Universal Visual Perception

autocut

bert-loves-chemistry

bert-loves-chemistry: a repository of HuggingFace models applied on chemical SMILES data for drug design, chemical modelling, etc.

best_AI_papers_2021

A curated list of the latest breakthroughs in AI (in 2021) by release date with a clear video explanation, link to a more in-depth article, and code.

Coacher

consistency_models

Official repo for consistency models.

CoOp

Learning to Prompt for Vision-Language Models.

Cream

This is a collection of our NAS and Vision Transformer work.

DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

DIG

A library for graph deep learning research

G-SchNet

G-SchNet - a generative model for 3d molecular structures

GeoMol

Graph2SMILES

Grounded-Segment-Anything

Marrying Grounding DINO with Segment Anything & Stable Diffusion & Tag2Text & BLIP & Whisper & ChatBot - Automatically Detect , Segment and Generate Anything with Image, Text, and Audio Inputs

HoiTransformer

This is the code for HOI Transformer

LWM

mage

A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis

minisora

The Mini Sora project aims to explore the implementation path and future development direction of Sora.

MobileSAM

This is the offiicial code for Faster Segment Anything (MobileSAM) project that makes SAM lightweight

MolRep

MolRep: A Deep Representation Learning Library for Molecular Property Prediction

Oscar

Oscar and VinVL

Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

SGG_from_NLS

Code repository for our paper "Learning to Generate Scene Graph from Natural Language Supervision" in ICCV 2021.

SinSR

[CVPR 2024] SinSR: Diffusion-Based Image Super-Resolution in a Single Step

smiles-transformer

Original implementation of the paper "SMILES Transformer: Pre-trained Molecular Fingerprint for Low Data Drug Discovery" by Shion Honda et al.

solvation-analysis

A set of functions and analysis classes for solvation structure analysis

stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

x-stable-diffusion

Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention.