用Transformer解决数论等整数问题,开源可扩展。
Int2Int: a framework for mathematics with transformers
- 基于PyTorch构建,支持数学对象的生成与解码
- 提供完整训练评估流程与数据处理工具
- 适合研究整数相关问题的开发者与数学科研者
本文介绍Int2Int,一个用于数学研究(特别是数论及其他整数相关问题)的开源Transformer代码库。Int2Int是完整的PyTorch实现,包含Transformer架构、训练与评估循环,以及用于表示、生成和解码常见数学对象的类与函数。附带数据准备脚本和可视化实验结果的Jupyter Notebook。本文详述Int2Int主要功能,作为用户手册,并提供扩展指南。Int2Int以MIT许可证发布,地址为https://github.com/f-charton/Int2Int。
原文摘要 · Abstract (English)
This paper documents Int2Int, an open source code base for using transformers on problems of mathematical research, with a focus on number theory and other problems involving integers. Int2Int is a complete PyTorch implementation of a transformer architecture, together with training and evaluation loops, and classes and functions to represent, generate and decode common mathematical objects. Ancillary code for data preparation, and Jupyter Notebooks for visualizing experimental results are also provided. This document presents the main features of Int2Int, serves as its user manual, and provides guidelines on how to extend it. Int2Int is released under the MIT licence, at https://github.com/f-charton/Int2Int.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。