Enable JIT compilation of the graph.

Cianciosa, Markrequested to merge
jit into main

This enables a working metal and Cuda jit compilation of the graph to enable efficient use of GPU resources.

Merge request reports