MIT License
Hierarchical Sketch Induction for Paraphrase Generation (Hosking et al., ACL 2022)
[ECCV2022] New benchmark for evaluating pre-trained model; New supervised contrastive learning fr...
Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023
Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for ...
CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch
Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding...
Implementation of Alphafold 3 in Pytorch
Chain-of-Hindsight, A Scalable RLHF Method
Multi-modal Molecule Structure-text Model for Text-based Editing and Retrieval, Nat Mach Intell 2...
Train vision models using JAX and 🤗 transformers
GLM (General Language Model)
A repository with exploration into using transformers to predict DNA ↔ transcription factor binding
Embed arbitrary modalities (images, audio, documents, etc) into large language models.