2 items across the graph — tagged with Tpu.
Large-scale LLM inference engine
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 interoperabil…