Highlights
- Pro
Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
TensorFlow code and pre-trained models for BERT
PyTorch implementations of Generative Adversarial Networks.
State-of-the-Art Text Embeddings
End-to-End Object Detection with Transformers
Open source code for AlphaFold 2.
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
A new kind of Progress Bar, with real-time throughput, ETA, and very cool animations!
Benchmarks of approximate nearest neighbor libraries in Python
PyTorch deep learning projects made easy.
Model summary in PyTorch similar to `model.summary()` in Keras
Evolutionary Scale Modeling (esm): Pretrained language models for proteins
PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
Siamese and triplet networks with online pair/triplet mining in PyTorch
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
Papers and Datasets about Point Cloud.
On the Variance of the Adaptive Learning Rate and Beyond
A curated list of community detection research papers with implementations.
Graphormer is a general-purpose deep learning backbone for molecular modeling.
An unofficial styleguide and best practices summary for PyTorch
PyTorch extensions for fast R&D prototyping and Kaggle farming
Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.
Pytorch implementation of Center Loss
allRank is a framework for training learning-to-rank neural models based on PyTorch.