JIT/GPU accelerated deep learning for Elixir with Axon v0.1