How do I convert a PyTorch model.parameters() to a JAX PyTree?

[NightMachinery]

import torch
from torchvision.models import resnet18, ResNet18_Weights
model = resnet18(weights=ResNet18_Weights.DEFAULT)
model.parameters() # how to get a PyTree out of this?!