tulvgengenr

Awesome-Bio-Foundation-Models

A collection of awesome bio-foundation models, including protein, RNA, DNA, gene, single-cell, and so on.

nnscaler

AlphaFold3

Implementation of Alpha Fold 3 from the paper: "Accurate structure prediction of biomolecular interactions with AlphaFold3" in PyTorch

alphafold3-pytorch

Implementation of Alphafold 3 in Pytorch

k-diffusion

Karras et al. (2022) diffusion models for PyTorch

diffusion-forcing

code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

fastllm

纯c++的全平台llm加速库，支持python调用，chatglm-6B级模型单卡可达10000+token / s，支持glm, llama, moss基座，手机端流畅运行

tianshou

An elegant PyTorch deep reinforcement learning library.

DPM-Solver-v3

Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

dpm-solver

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

PiPPy

Pipeline Parallelism for PyTorch

awesomeMLSys

An ML Systems Onboarding list

lectures

Material for cuda-mode lectures

algorithms

Algorithms & Data structures in C++.

CUDATutorial

A CUDA tutorial to make people learn CUDA program from 0

CUDA-Programming-Guide-in-Chinese

This is a Chinese translation of the CUDA programming guide

CUDA-Learn-Notes

🎉CUDA/C++ 笔记 / 大模型手撕CUDA / 技术博客，更新随缘: flash_attn、sgemm、sgemv、warp reduce、block reduce、dot product、elementwise、softmax、layernorm、rmsnorm、hist etc.

poseidon

Code for the paper "Poseidon: Efficient Foundation Models for PDEs"

ai4sciencetalks.github.io

SwiftTransformer

High performance Transformer implementation in C++.

mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

edm

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

concode

Mapping Language to Code in a Programmatic Context

CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation

CodeXGLUE

CodeXGLUE

get-started-with-JAX

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

ddpo-pytorch

DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support