JAX Adds Custom Opaque Types Enabling Differentiable Rasterization

srush_nlp · x · 2026-08-01

Sasha Rush highlighted a new feature in the JAX framework: support for opaque types with custom tangents.

A cool use case for this feature is the proper handling of batched geometric objects. The thread includes a short demo of differentiable rasterization, showing how to fit a photo using 1500 triangles by minimizing the L2 distance. The author noted that it is interesting to see differentiable computing move beyond traditional tensors.

Related event: JAX Introduces Custom Opaque Types for Differentiable Rendering(2 posts)→

Original post →

More from Research

Research channel →