TorchX is a library containing standard DSLs for authoring and running PyTorch related components for an E2E production ML pipeline - View it on GitHub