Geek Repo
followers
following
stars
Github PK Tool:Github PK Tool
functorch is a prototype of JAX-like composable function transforms for PyTorch.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Open deep learning compiler stack for cpu, gpu and specialized accelerators