From PyTorch to JAX# The following tutorials provide an onboarding path to JAX, for users who are familiar with PyTorch: JAX for PyTorch users Porting a PyTorch model to JAX