Trending

PyTorch Helion

Leaving Meta and PyTorch

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

Meta's Optimization Platform Ax 1.0 Streamlines LLM and System Optimization

Now stable, Ax is an open-source platform from Meta designed to help researchers and engineers apply machine learning to complex, resource-intensive experimentation. Over the past several years, Meta has used Ax to improve AI models, accelerate machine learning research, tune production infrastructure, and more. By Sergio De Simone

[P] CleanMARL : a clean implementations of Multi-Agent Reinforcement Learning Algorithms in PyTorch

[P] Built a differentiable parametric curves library for PyTorch

[D] Do you ever miss PyTorch-style workflows?

[P] DCNv2 (Update Compatibility) Pytorch 2.8.0

[P] Reproducing YOLOv1 From Scratch in PyTorch - Learning to Implement Object Detection from the Original Paper

[D] problems with pytorch's mps backend

[D] What is the best code assistant to use for PyTorch?

Inside Google Gemma 3n: my PyTorch Profiler insights

[R] kappaTune: a PyTorch-based optimizer wrapper for continual learning via selective fine-tuning

DLR-RM

Save item

stable-baselines3

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

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.

microsoft

Save item

onnxruntime

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

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.

speechbrain

Save item

speechbrain

A PyTorch-based Speech Toolkit

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