Support numpy/jax

[srush]

My summer project is to port all the layout code to use Numpy. This mainly will mean rewriting V2, P2, Affine, colors to be in matrix representation.

• I think this is mathematically cleaner, and shouldn't require changing the user API
• It will allow "batched diagrams". Not sure that is useful for anything, but maybe it is a cool way to make animations (if the tree is the same)?
• It will allow differentiating though the rendering pipeline. This doesn't seem useful, but it is really fun.

[danoneata]

Cool! Sounds great! 🙂 I've also started here an animation repository; and I will be porting there some of the Haskell code: the active library and the diagrams.animation libraries.