Nemotron-SWE-v1 est un grand ensemble de données affiné par instructions, orienté vers les agents de génie logiciel, publié par NVIDIA en 2025. Les travaux de recherche associés sont présentés dans « Training Software Engineering Agents and Verifiers with SWE-Gym ». Il vise à améliorer la compréhension du code, la planification multi-étapes et les capacités de correction des grands modèles de langage dans le cadre réel du cycle de vie du développement logiciel.

L'ensemble de données est basé sur les descriptions de problèmes issues de SWE-Gym et R2E-Gym-Subset. Les trajectoires d'interaction des agents ont été synthétisées à l'aide du modèle Qwen3-Coder-480B-A35B-Instruct via le framework OpenHands. Il est principalement destiné à l'ajustement supervisé (SFT), aidant les modèles à traiter des tâches de génie logiciel similaires à celles de SWE-Bench.

Composition de l'ensemble de données

L'ensemble de données comprend un sous-ensemble r2e_gym, contenant principalement les champs suivants :

• messages : enregistre la séquence de messages des interactions entre l'agent et l'environnement, incluant la description de la tâche, les réponses du modèle et le contenu des interactions avec les outils.
• license : enregistre les informations de licence correspondant à chaque échantillon.
• used_in : enregistre les modèles ou projets dans lesquels l'échantillon a été utilisé.
• tools : enregistre les outils accessibles à l'agent ainsi que leurs définitions.
• dataset : enregistre la source de l'ensemble de données original de l'échantillon.
• repo : enregistre le nom du dépôt de code associé.