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.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-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-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.pytorch-debug-assistant