axon
axon copied to clipboard
Add multi-device training/inference
Requires https://github.com/elixir-nx/nx/issues/645