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

Image and Keypoints Augmentations in Jax transpiled Kornia

In this demo, we apply augmentations to images using Jax transpiled Kornia

Denoise Image using TensorFlow transpiled Kornia

In this demo, we denoise an image using TensorFlow transpiled Kornia