Allow for different backends #13
Description
Ideally it should be possible to have multiple backends (PyTorch, Jax) and to be able to switch easily between them.
Currently only PyTorch is supported.
{{ message }}
Ideally it should be possible to have multiple backends (PyTorch, Jax) and to be able to switch easily between them.
Currently only PyTorch is supported.