Junzhe Li's starred repositories
Awesome-Bio-Foundation-Models
A collection of awesome bio-foundation models, including protein, RNA, DNA, gene, single-cell, and so on.
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"
DPM-Solver-v3
Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)
dpm-solver
Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)
awesomeMLSys
An ML Systems Onboarding list
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.
SwiftTransformer
High performance Transformer implementation in C++.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
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