Comment on Jax's true calling: Ray-Marching renderers on WebGLComments−chillee5moI'd also note that you can more or less write the same code in PyTorch with torch.vmap
Comments
I'd also note that you can more or less write the same code in PyTorch with torch.vmap