> the thing JAX was truly meant for: a graphics renderer<p>I mean, just like ray-tracing, SDF (ray-marching) is neat, but basically everything useful is expensive or hard to do (collisions, meshes, texturing etc.). I mean mathy stuff is easier (rotations, unions/intersections, function composition, etc.) but 3D is usually used in either modeling software or video games, which care more about the former than they do the latter.
Games and simulations are typically stateful, I'm not sure the functional purity of JAX is a good fit.<p>Also, what's the story for JAX + WebGL when it comes to targeting hardware-accelerated ray tracing?