Installation

Installation depends on torch, jax and numpy packages.

To install, simply run

$ pip install git+https://github.com/rdyro/torch2jax.git