A Multi-agent AI System for Deep Learning Model Migration from TensorFlow to JAX

ArXi:2603.27296v1 Announce Type: cross The rapid development of AI-based products and their underlying models has led to constant innovation in deep learning frameworks. Google has been pioneering machine learning usage across dozens of products. Maintaining the multitude of model source codes in different ML frameworks and versions is a significant challenge. So far the maintenance and migration work was done largely manually by human experts. We describe an AI-based multi-agent system that we built to automatic migration of TensorFlow-based deep learning models into JAX-based ones.