Software Engineer at Google
-
Google
- Kirkland, WA
- https://sites.google.com/view/jialei
Stars
torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytor…