GitHub - unifyai/ivy: The Unified Machine Learning Framework

This example uses PyTorch as a backend framework, but the backend can easily be changed to your favorite frameworks, such as TensorFlow, JAX, or MXNet.