xfcui's Stars
facebookresearch/faiss
A library for efficient similarity search and clustering of dense vectors.
huggingface/pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
pyg-team/pytorch_geometric
Graph Neural Network Library for PyTorch
ml-tooling/best-of-ml-python
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
Dao-AILab/flash-attention
Fast and memory-efficient exact attention
state-spaces/mamba
Mamba SSM architecture
jacobgil/pytorch-grad-cam
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
facebookresearch/pytorch3d
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
deepchem/deepchem
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials Science and Biology
jettify/pytorch-optimizer
torch-optimizer -- collection of optimizers for Pytorch
rdkit/rdkit
The official sources for the RDKit library
davda54/sam
SAM: Sharpness-Aware Minimization (PyTorch)
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
DeepGraphLearning/torchdrug
A powerful and flexible machine learning platform for drug discovery
jax-md/jax-md
Differentiable, Hardware Accelerated, Molecular Dynamics
e3nn/e3nn
A modular framework for neural networks with Euclidean symmetry
xxxnell/how-do-vits-work
(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
sail-sg/Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
ccsb-scripps/AutoDock-Vina
AutoDock Vina
ict-bigdatalab/awesome-pretrained-models-for-information-retrieval
A curated list of awesome papers related to pre-trained models for information retrieval (a.k.a., pretraining for IR).
torchmd/torchmd
End-To-End Molecular Dynamics (MD) Engine using PyTorch
facebookresearch/msn
Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
torchmd/torchmd-net
Training neural network potentials
torchmd/mdgrad
Pytorch differentiable molecular dynamics
LingxiaoShawn/GNNAsKernel
Source code for From Stars to Subgraphs (ICLR 2022)