3 ms·
how gpu acceleration works under the hood ? There is no such info in the blogpost.
by m00dy 4y ago
how gpu acceleration works under the hood ? There is no such info in the blogpost.
- throwawaymaths 4y agoThere's a library called Nx which abstracts elixir code into backend engines like xla. You could write your own backend too if you wanted. Edit: xla, not jax
- josevalim 4y agoIt uses [Nx](https://github.com/elixir-nx/nx/tree/main/nx#readme https://github.com/elixir-nx/nx/tree/main/nx#readme) (think of it as an equivalent to numpy/jax) which uses Google XLA's as one of the possible backends to compile and run on the GPU.