datasets
Training and evaluation data, with the modality, task and licence stated up front. Listed live from the Hugging Face Hub.
Pytorch-Code-10K
Python/Pytorch Code Dataset
A collection of code from repos on The Stack, with captions generated by AI.
Dataset Description
This dataset contains Python code snippets sourced from open-source repositories that utilize PyTorch or Hugging Face Transformers. Each sample includes:
code: The raw Python source code (typically containing import torch, from torch import nn, or transformer-related imports)
caption: A natural language description generated by T5-Large… See the full description on the dataset page: https://huggingface.co/datasets/Monster-Code/Pytorch-Code-10K.retailhero-uplifthttps://ods.ai/competitions/x5-retailhero-uplift-modeling
transactions-genderhttps://www.kaggle.com/c/python-and-analyze-data-final-project/
pytorch-issues-dataset-cleanPyTorchConference2025_GithubRepos
PyTorch Conference 2025 GitHub Repos
I created a list of every GitHub repo mentioned during PyTorch Conference 2025 and Open Source AI Week.
gpu-cuda-pytorch-compatibility
PyTorch release, CUDA build and NVIDIA driver pairings
Canonical, always-current version: https://referencesource.org/gpu-cuda-pytorch-compatibility/
Machine-readable: https://referencesource.org/gpu-cuda-pytorch-compatibility/data.json — this mirror is a point-in-time copy.
Last verified: 2026-10-01
Stale after: 2026-12-28 (past this date, prefer the canonical copy —
it re-verifies on a cadence this snapshot does not)
Records: 45
One record per (PyTorch release, CUDA build)… See the full description on the dataset page: https://huggingface.co/datasets/referencesource/gpu-cuda-pytorch-compatibility.rosbank-churnhttps://boosters.pro/championship/rosbank1/
github-pytorch-issues
Dataset Card for github-pytorch-issues
Dataset Summary
This dataset is a curated collection of GitHub issues from the PyTorch repository. Each entry includes the issue title, body, user, state, labels, comments, and other relevant fields that are useful for tasks such as text classification, semantic search, and question answering.
Supported Tasks and Leaderboards
The dataset supports the following tasks:
Open-domain Question Answering: Given a user query… See the full description on the dataset page: https://huggingface.co/datasets/mayankpuvvala/github-pytorch-issues.pytorch-issues
Dataset Card for "pytorch-issues"
More Information needed
pytorch-semantic-dataset-fixed
PyTorch Semantic Code Dataset
A semantically-enriched Python code dataset combining syntactic tokenization with deep semantic analysis from Language Server Protocol (LSP) tools.
🎯 Overview
This dataset enhances tokenized Python code with semantic embeddings derived from static analysis tools (Tree-sitter + Jedi), providing models with both syntactic and semantic understanding of code symbols. Each token in the code is aligned with rich semantic information including type… See the full description on the dataset page: https://huggingface.co/datasets/ant-des/pytorch-semantic-dataset-fixed.ttrshttps://arxiv.org/abs/2110.05589
pytorch-tasks-v1
pytorch-tasks-v1
Task dataset for a PyTorch RL / eval environment, in the shape used by the
Prime Intellect Environments Hub.
45 PyTorch tasks across 5 categories. Each task gives the model one or more input tensors and
an instruction; the answer is the tensor left in result, graded with torch.allclose
against a reference. Grading is deterministic — no LLM judge, no external API, CPU only.
Category
Tasks
Covers
tensor_ops
16
reshape, transpose, axis reductions, clamp… See the full description on the dataset page: https://huggingface.co/datasets/eltociear/pytorch-tasks-v1.kkbox
Dataset Card for "kkbox"
More Information needed
datascience-bowl2019https://www.kaggle.com/c/data-science-bowl-2019
pytorch_assistant
Dataset Card for pytorch_assistant
This dataset has been created with distilabel.
Dataset Summary
This dataset contains a pipeline.yaml which can be used to reproduce the pipeline that generated it in distilabel using the distilabel CLI:
distilabel pipeline run --config "https://huggingface.co/datasets/dvilasuero/pytorch_assistant/raw/main/pipeline.yaml"
or explore the configuration:
distilabel pipeline info --config… See the full description on the dataset page: https://huggingface.co/datasets/dvilasuero/pytorch_assistant.acquire-valued-shoppershttps://www.kaggle.com/c/acquire-valued-shoppers-challenge
pytorch-tutorial-168
Dataset Card for "pytorch-tutorial-168"
More Information needed
pytorch-discuss-tutorial-1000
Dataset Card for "pytorch-discuss-tutorial-1000"
More Information needed
pytorch-forum-topics-complete-v2
PyTorch Forum Topics Dataset
This dataset contains topic metadata scraped from the PyTorch Community Forum. It includes comprehensive information about forum topics that can be used for various NLP tasks related to PyTorch and deep learning discussions.
Dataset Structure
Each record in the dataset contains the following fields:
id: Unique topic identifier
title: Topic title
slug: URL-friendly version of the title
posts_count: Number of posts in the topic
reply_count:… See the full description on the dataset page: https://huggingface.co/datasets/AmitPrakash/pytorch-forum-topics-complete-v2.test-pytorch
Dataset Card for test-pytorch
This dataset has been created with distilabel.
Dataset Summary
This dataset contains a pipeline.yaml which can be used to reproduce the pipeline that generated it in distilabel using the distilabel CLI:
distilabel pipeline run --config "https://huggingface.co/datasets/dvilasuero/test-pytorch/raw/main/pipeline.yaml"
or explore the configuration:
distilabel pipeline info --config… See the full description on the dataset page: https://huggingface.co/datasets/dvilasuero/test-pytorch.annotated_dataset_o3_train_pytorch_tritonstack_overflow_pytorchgene_annotationspytorch-discuss-tutorial-346
Dataset Card for "pytorch-discuss-tutorial-346"
More Information needed
pytorch-debug-assistantmedical_spanish_pytorchpytorch-reasoningSamples in this benchmark were generated by RELAI using the following data source(s):
Data Source Name: pytorch
Data Source Link: https://pytorch.org/docs/stable/index.html
Data Source License: https://github.com/pytorch/pytorch/blob/main/LICENSE
Data Source Authors: PyTorch
AI Benchmarks by Data Agents. 2025 RELAI.AI. Licensed under CC BY 4.0. Source: https://relai.ai
tinkoff-data-likeTinkoff Data-Like 2019 competition dataset
pytorchmodelmetadata_with_embeddingspytorch_fn_sft_test