Skip to main content

torch2jax:旨在实现在JAX中无需拷贝即可调用PyTorch代码,无论是在即时执行还是即时编译(JIT)模式下