Cover Image

Learning Bug Context for PyTorch-to-JAX Translation with LLMs

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Phan, Hung, Vu, Son Le, Jannesari, Ali
Format:	Preprint
Published:	2025
Subjects:	Machine Learning Artificial Intelligence
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

torchtune: PyTorch native post-training library
by: Obozov, Mark, et al.
Published: (2026)

PiShield: A PyTorch Package for Learning with Requirements
by: Stoian, Mihaela Cătălina, et al.
Published: (2024)

TiledAttention: a CUDA Tile SDPA Kernel for PyTorch
by: Khan, Taimur
Published: (2026)

HAT-CL: A Hard-Attention-to-the-Task PyTorch Library for Continual Learning
by: Duan, Xiaotian
Published: (2023)

depyf: Open the Opaque Box of PyTorch Compiler for Machine Learning Researchers
by: You, Kaichao, et al.
Published: (2024)

Fully Differentiable dMRI Streamline Propagation in PyTorch
by: Yoon, Jongyeon, et al.
Published: (2025)

Soft Decision Tree classifier: explainable and extendable PyTorch implementation
by: Shamir, Reuben R
Published: (2025)

pyFAST: A Modular PyTorch Framework for Time Series Modeling with Multi-source and Sparse Data
by: Wang, Zhijin, et al.
Published: (2025)

Demystifying the Silence of Correctness Bugs in PyTorch Compiler
by: Li, Meiziniu, et al.
Published: (2026)

TorchTitan: One-stop PyTorch native solution for production ready LLM pre-training
by: Liang, Wanchao, et al.
Published: (2024)

KHNNs: hypercomplex neural networks computations via Keras using TensorFlow and PyTorch
by: Niemczynowicz, Agnieszka, et al.
Published: (2024)

MIP Candy: A Modular PyTorch Framework for Medical Image Processing
by: Fu, Tianhao, et al.
Published: (2026)

Comparative Analysis of CNN Performance in Keras, PyTorch and JAX on PathMNIST
by: Nezović, Anida, et al.
Published: (2025)

PyLO: Towards Accessible Learned Optimizers in PyTorch
by: Janson, Paul, et al.
Published: (2025)

Hessian of Perplexity for Large Language Models by PyTorch autograd (Open Source)
by: Ilin, Ivan
Published: (2025)

Performance comparison of medical image classification systems using TensorFlow Keras, PyTorch, and JAX
by: Bećirović, Merjem, et al.
Published: (2025)

Leveraging Manifold Embeddings for Enhanced Graph Transformer Representations and Learning
by: Jyothish, Ankit, et al.
Published: (2025)

LTNtorch: PyTorch Implementation of Logic Tensor Networks
by: Carraro, Tommaso, et al.
Published: (2024)

TorchAO: PyTorch-Native Training-to-Serving Model Optimization
by: Or, Andrew, et al.
Published: (2025)

A Comparative Survey of PyTorch vs TensorFlow for Deep Learning: Usability, Performance, and Deployment Trade-offs
by: Alawi, Zakariya Ba
Published: (2025)

PyGraph: Robust Compiler Support for CUDA Graphs in PyTorch
by: Ghosh, Abhishek, et al.
Published: (2025)

Venom: A PyTorch Generative Modeling Toolkit
by: Yan, Liang
Published: (2026)

Lowering PyTorch's Memory Consumption for Selective Differentiation
by: Bhatia, Samarth, et al.
Published: (2024)

torchgfn: A PyTorch GFlowNet library
by: Viviano, Joseph D., et al.
Published: (2023)

Torch Geometric Pool: the PyTorch library for pooling in Graph Neural Networks
by: Abate, Carlo, et al.
Published: (2025)

ExecuTorch -- A Unified PyTorch Solution to Run AI Models On-Device
by: Nachin, Mergen, et al.
Published: (2026)

NewsTorch: A PyTorch-based Toolkit for Learner-oriented News Recommendation
by: Wang, Rongyao, et al.
Published: (2026)

A PyTorch Library of Turing-Complete Neural Networks
by: Bates, Jonathan
Published: (2026)

torchmil: A PyTorch-based library for deep Multiple Instance Learning
by: Castro-Macías, Francisco M., et al.
Published: (2025)

Flashlight: PyTorch Compiler Extensions to Accelerate Attention Variants
by: You, Bozhi, et al.
Published: (2025)

FLoRA: Enhancing Vision-Language Models with Parameter-Efficient Federated Learning
by: Nguyen, Duy Phuong, et al.
Published: (2024)

CrypTorch: PyTorch-based Auto-tuning Compiler for Machine Learning with Multi-party Computation
by: Liu, Jinyu, et al.
Published: (2025)

PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning
by: Hu, Weihua, et al.
Published: (2024)

A PyTorch Framework for Scalable Non-Crossing Quantile Regression
by: Chang, Kaihua
Published: (2025)

Torch-Choice: A PyTorch Package for Large-Scale Choice Modeling with Python
by: Du, Tianyu, et al.
Published: (2023)

Studying the Impact of TensorFlow and PyTorch Bindings on Machine Learning Software Quality
by: Li, Hao, et al.
Published: (2024)

Fuzzy PyTorch: Rapid Numerical Variability Evaluation for Deep Learning Models
by: Gonzalez-Pepe, Inés, et al.
Published: (2026)

PuzzleJAX: A Benchmark for Reasoning and Learning
by: Earle, Sam, et al.
Published: (2025)

VMAF Re-implementation on PyTorch: Some Experimental Results
by: Aistov, Kirill, et al.
Published: (2023)

Federated Foundation Models: Privacy-Preserving and Collaborative Learning for Large Models
by: Yu, Sixing, et al.
Published: (2023)