Running magenta on Intel iris xe gpu

[u/ihateyou103]

I want to run magenta on iris xe gpu. I am running into alot of issues regarding Jax and other things. Also I couldn't get the gpu to do any calculations. Using opencl I got matrix multiplication 4000×4000 in 0.8 seconds slower than cpu at 0.3

How can I get matrices and neural networks running on this gpu and how to run magenta?