Migration includes training and delivered weights
XStack, Princeton University’s benchmark for migrating machine learning repositories, requires more than rewritten code. Agents move projects built with PyTorch, a machine learning framework for graphics processors, to tensor processing units using the JAX software ecosystem. Twenty repositories cover fourteen domains. The assignment includes installing dependencies, preparing data, training, evaluating and exporting weights that can be loaded by the original project.[1]
