pytorch

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

Machine learning metrics for distributed, scalable PyTorch applications.

:computer:深度学习实战：手写数字识别、Discuz验证码识别、垃圾分类、语义分割

Medical imaging processing for AI applications.

Türkiye'de yapılan derin öğrenme (deep learning) ve makine öğrenmesi (machine learning) çalışmalarının derlendiği sayfa.

Large-scale pretraining for dialogue

LeetCode for PyTorch — 65 ML/AI interview problems from real interviews at Google, Meta, Anthropic. Jupyter notebooks, an auto-grader, and an MCP AI tutor.

pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.

Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch

A powerful tool that translates ComfyUI workflows into executable Python code.

HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis

Objectron is a dataset of short, object-centric video clips. In addition, the videos also contain AR session metadata including camera poses, sparse point-clouds and planes. In each video, the camera moves around and above the object and captures it from different views. Each object is annotated with a 3D bounding box. The 3D bounding box describes the object’s position, orientation, and dimensions. The dataset contains about 15K annotated video clips and 4M annotated images in the following categories: bikes, books, bottles, cameras, cereal boxes, chairs, cups, laptops, and shoes

Run Mixtral-8x7B models in Colab or consumer desktops

🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM, Finer-CAM, LeGrad, RefineCAM)

ICCV2019 - Learning to Paint With Model-based Deep Reinforcement Learning

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".

Deep learning PyTorch library for time series forecasting, classification, and anomaly detection (originally for flood forecasting).

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

A library for training and deploying machine learning models on Amazon SageMaker