Compiling Jax for FPGA?

[rog77]

This is more a question of interest rather than need (would otherwise be using TPU), but is this possible?

I read here it is possible to export HLO IR from jax - http://www.bnikolic.co.uk/blog/python/jax/2022/02/22/jax-outputgraph-rev.html

And this tool chain will convert TF/XLA to verilog: https://github.com/danielholanda/LeFlow

I am interested in hearing anyone's thoughts on how plausible this is before I go off down a rabbit hole!

Just ordered a De-10 nano, and would like to do something other than play retrogames when it arrives... Board has arm cores to run Linux, and hopefully the fpga side can take the place of a GPU/TPU?

Thanks in advance.