datasets
Training and evaluation data, with the modality, task and licence stated up front. Listed live from the Hugging Face Hub.
Pytorch-Code-10K
Hot Coco Training Dataset
A curated collection of 10,625 high-quality PyTorch and Transformers code examples with AI-generated captions. This dataset was specifically built for fine-tuning code-specialized language models like Qimi (Coming soon!)
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… 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
pytorch-issues-dataset-cleantransactions-genderhttps://www.kaggle.com/c/python-and-analyze-data-final-project/
PyTorchConference2025_GithubRepos
PyTorch Conference 2025 GitHub Repos
I created a list of every GitHub repo mentioned during PyTorch Conference 2025 and Open Source AI Week.
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.rosbank-churnhttps://boosters.pro/championship/rosbank1/
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.acquire-valued-shoppershttps://www.kaggle.com/c/acquire-valued-shoppers-challenge
datascience-bowl2019https://www.kaggle.com/c/data-science-bowl-2019
ttrshttps://arxiv.org/abs/2110.05589
pytorchmodelmetadata_with_embeddingspytorch-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.stack_overflow_pytorchtinkoff-data-likeTinkoff Data-Like 2019 competition dataset
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.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.pytorch-discuss-tutorial-1000
Dataset Card for "pytorch-discuss-tutorial-1000"
More Information needed
pytorch-tutorial-168
Dataset Card for "pytorch-tutorial-168"
More Information needed
annotated_dataset_o3_train_pytorch_tritonpytorch-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
kkbox
Dataset Card for "kkbox"
More Information needed
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.pytorch-discuss-tutorial-346
Dataset Card for "pytorch-discuss-tutorial-346"
More Information needed
gene_annotationspytorch-debug-assistantpytorch-standardSamples 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
pytorchmodelmetadatapytorch_scrape_inductor_dataThis dataset is composed of scraping code off of github containing pytorch code, and then running torch compile on it in order to have pairs of pytorch and triton code.
To spot check the data you can run
from datasets import load_dataset
# Load the dataset
ds = load_dataset("GPUMODE/pytorch_scrape_inductor_data")
# Get the first row
row = ds["train"][0]
print("\n=== UUID ===")
print(row['uuid'])
print("\n=== PYTHON CODE ===")
print(row['python_code'])
print("\n=== TRITON CODE ===")… See the full description on the dataset page: https://huggingface.co/datasets/GPUMODE/pytorch_scrape_inductor_data.