Trending

[P] Tensorlink: A Framework for Model Distribution and P2P Resource Sharing in PyTorch

[D] How to train a model for food image classification in PyTorch? [D]

Intel Updates Its PyTorch Extension With DeepSeek-R1 Support, New Optimizations

Pytorch 2.7.0 with support for Blackwell (5090, B200) to come out today

[R] Can’t Train LoRA + Phi-2 on 2x GPUs with FSDP — Keep Getting PyArrow ArrowInvalid, DTensor, and Tokenization Errors

PSA: You can do QAT (quantization aware tuning) with Meta's torchtune.

[D] Using Pytorch GradScaler results in NaN weights

[P] Help required for a project using Pytorch Hooks

Kyutai

Save item

moshika-vis-pytorch-bf16

microsoft

Save item

torchgeo

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

huggingface

Save item

transformers

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

ultralytics

Save item

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

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

DLR-RM

Save item

stable-baselines3

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