RuiWang1998

numpy

The fundamental package for scientific computing with Python.

CASP15

CASP15 performance benchmarking of the state-of-the-art protein structure prediction methods

foldcomp

Compressing protein structures effectively with torsion angles

ml-sigma-reparam

cutlass

CUDA Templates for Linear Algebra Subroutines

tree

tree is a library for working with nested data structures

s4

Structured state space sequence models

mamba

float8_experimental

This repository contains the experimental PyTorch native float8 training UX

segment-anything-fast

A batched offline inference oriented version of segment-anything

opt_einsum

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

mistral-src

Reference implementation of Mistral AI 7B v0.1 model.

ProteinPretraining

Source code of PETA: Evaluating the Impact of Protein Transfer Learning with Sub-word Tokenization on Downstream Applications.

MS-AMP

Microsoft Automatic Mixed Precision Library

PEER_Benchmark

PEER Benchmark, appear at NeurIPS 2022 Dataset and Benchmark Track (https://arxiv.org/abs/2206.02096)

EATLM

Code for 'On Pre-trained Language Models For Antibody'

Sophia

The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”

ColabDesign

Making Protein Design accessible to all via Google Colab!

Sophia

Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.

MiniFold

MiniFold: Deep Learning for Protein Structure Prediction inspired by DeepMind AlphaFold algorithm

OmniQuant

[ICLR2024 spotlight] OmniQuant is a simple and powerful quantization technique for LLMs.

Baichuan-7B

A large-scale 7B pretraining language model developed by BaiChuan-Inc.

RoseTTAFold2

dadaptation

D-Adaptation for SGD, Adam and AdaGrad

LoRA

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

ControlNet

Let us control diffusion models!

llama

Inference code for Llama models

Ankh

Ankh: Optimized Protein Language Model

FlexGen

Running large language models on a single GPU for throughput-oriented scenarios.