AIBullisharXiv – CS AI · 11h ago6/10
🧠
Learning Bug Context for PyTorch-to-JAX Translation with LLMs
Researchers introduce T2J, a benchmark dataset of PyTorch-to-JAX translation bugs paired with developer fixes, addressing the challenge of translating deep-learning code between frameworks. By training LLMs on this curated bug-fix data through in-context learning, they achieve up to 20% improvement in translation accuracy, demonstrating that domain-specific bug datasets can significantly enhance code generation reliability.
🧠 GPT-4