Nx: Multi-dimensional tensors Elixir lib with multi-staged compilation (CPU/GPU)
Nx: Multi-dimensional tensors Elixir lib with multi-staged compilation (CPU/GPU)
Feb 17, 2021, 5:52pm UTC
https://github.com/elixir-nx/nx/tree/main/nx