subho406/Pytorch2Jax

Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.

2 best points · 0 comments · 1 Hacker News post · 21 GitHub stars · View on GitHub

Hacker News discussions