Trending

Leaving Meta and PyTorch

PyTorch Helion

The bug that taught me more about PyTorch than years of using it

Torchcomms: A modern PyTorch communications API

The State of Machine Learning Frameworks in 2019

PyTorch Monarch

A beginner's guide to deploying LLMs with AMD on Windows using PyTorch

I Solved PyTorch's Cross-Platform Nightmare

Speeding up PyTorch inference by 87% on Apple with AI-generated Metal kernels

Google

Save item

timesfm-2.5-200m-pytorch

Task: time-series-forecasting

Kyutai

Save item

moshika-vis-pytorch-bf16

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

huggingface

Save item

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

huggingface

Save item

transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

speechbrain

Save item

speechbrain

A PyTorch-based Speech Toolkit

microsoft

Save item

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

DLR-RM

Save item

stable-baselines3

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

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

pyg-team

Save item

pytorch_geometric

Graph Neural Network Library for PyTorch