ProteinDT

hrq-vae

Hierarchical Sketch Induction for Paraphrase Generation (Hosking et al., ACL 2022)

OmniBenchmark

[ECCV2022] New benchmark for evaluating pre-trained model; New supervised contrastive learning fr...

Geom3D

Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023

lossyless

Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for ...

cleanba

CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL

enformer-pytorch

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch

TUPE

Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding...

alphafold3-pytorch

Implementation of Alphafold 3 in Pytorch

chain-of-hindsight

Chain-of-Hindsight, A Scalable RLHF Method

MoleculeSTM

Multi-modal Molecule Structure-text Model for Text-based Editing and Retrieval, Nat Mach Intell 2...

clip-jax

Train vision models using JAX and 🤗 transformers

goat-bench

GLM

GLM (General Language Model)

tf-bind-transformer

A repository with exploration into using transformers to predict DNA ↔ transcription factor binding

multi_token

Embed arbitrary modalities (images, audio, documents, etc) into large language models.