Stars
Browse starred repositories and topics
Sort: Recently starred
Starred Repositories
-
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
-
Protein structure diffusion model for unconditional protein generation and motif scaffolding
-
generalized machine learning potential eqnorm
-
-
Code for the ProteinMPNN paper
-
A bunch of kernels that might make stuff slower 😉
-
Minimal and annotated implementations of key ideas from modern deep learning research.
-
Stronger ViTs With Octic Equivariance
-
Efficient 3D molecular generation with flow-matching and Semla
-
Source-to-Source Debuggable Derivatives in Pure Python