You can think of it as an optimized & extended NumPy that compiles to any of JAX, TF, PyTorch, or… NumPy itself.
NumPy optimization framework supporting multiple deep learning backends
By
–
By
–
You can think of it as an optimized & extended NumPy that compiles to any of JAX, TF, PyTorch, or… NumPy itself.