JAX’s true calling: Ray-Marching renderers in Python on WebGL Demo (move your mouse/thumb across the image) Why, though? Well, I’ve been drooling over this tool the cool kids use, and wondering how I can join the gang. It’s called JAX. It’s got GPU accelerated functions over n-dimensional arrays. And built-in compile-time differentiability of these!? Auto-vectorization?? And you just have to do like with numpy. What’s not to like? Go home APL!