Early work fleshing out key potential pain points in using JAX as the framework within which to build a Python EGSnrc port.