Torch 환경에서 작성된 코드를 Jax로 run 하기 위하여
Torch2Jax 관련 소스를 살펴보았습니다.
나중엔 Torch 환경 코드를 Jax환경에 맞춰
port를 할 예정이라 참고하기 좋을 것 같아 정리하기로 했습니다.
https://github.com/rdyro/torch2jax
https://robertdyro.com/torch2jax/examples/resnet_example/
I'm an graduate student majoring in Computer Engineering at Inha University. I'm interested in Machine learning developing frameworks, Formal verification, and Concurrency.