Trending

PyGraph: Robust Compiler Support for CUDA Graphs in PyTorch

Show HN: Keep your PyTorch model in VRAM by hot swapping code

PyTorch Internals: Ezyang's Blog

Show HN: Torch Lens Maker – Differentiable Geometric Optics in PyTorch

Kyutai

Save item

moshika-vis-pytorch-bf16

huggingface

Save item

transformers

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

lucidrains

Save item

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

huggingface

Save item

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

ultralytics

Save item

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

microsoft

Save item

torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

DLR-RM

Save item

stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.