GitHub repository: lucidrains/jax2torch
Primary language: Python
Description: Use Jax functions in Pytorch