Examples and Demos

Accelerating PyTorch models with JAX

Accelerate your Pytorch models by converting them to JAX for faster inference.

Finetuning PyTorch Models in your TensorFlow Projects

In this demo, we finetune the PyTorch ResNet model in TensorFlow