nx
nx copied to clipboard
elixir-nx
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
The key which is based on mix.lock fails to point to the correct file (probably due to the working_directory change).
- Implementation of stateless PRNG, based on splittable keys and the Threefry algorithm - Based on the JAX PRNG implementation and yield the same results - At the moment includes...
Reference JAX implementation: https://github.com/google/jax/pull/1749
GPU is a GTX1060 Driver Version: 515.48.07 CUDA Version: 11.7 `$ XLA_TARGET=cuda111 XLA_TARGET=cuda mix run -e " EXLA.jit(&( &1 ), [Nx.tensor([1])])"` ``` 15:53:47.914 [error] GenServer EXLA.Client terminating ** (RuntimeError) Could...
See the tensorflow feature and linked paper: https://www.tensorflow.org/api_docs/python/tf/vectorized_map
Currently EXLA transfer operations block until the transfer is complete by using a mixture of `kImmutableUntilTransferCompletes` and then blocking the host before returning from the NIF. However, I believe we...
We need to add support for complex numbers for the function and for it's grad implementation - [ ] BinaryBackend - [ ] EXLA - [ ] Torchx - [...
We need to add support for complex numbers for the function and for it's grad implementation - [ ] BinaryBackend - [ ] EXLA - [ ] Torchx - [...
To enable at most once-copy implementations from Arrow backends.
As discussed in #331, we need an approach for having portable PRNGs that can be applied for any backend. > We definitely need the ability to seed RNGs, and I'm...