Haskell and Tensorflow

Learn
3

+1 on having Jax bindings. I would try to help, if I get some time. Also, there are bindings to XLA already, as a starting point: Haskell and XLA

Hasktorch is binding to PyTorch.