/jax_v_pytorch

comparing Jax (haiku and flax) to PyTorch on simple DL tasks.

Primary LanguagePythonMIT LicenseMIT

Stargazers