jax2torch

Python ★ 263 updated 3y ago

Use Jax functions in Pytorch

No plain-English explanation yet — one is being written right now. Check back in a minute.