Support numpy/jax - Githubissues

chalk-diagrams / chalk

A declarative drawing API in Python

MIT License

275 stars 13 forks source link

Support numpy/jax #133

Open srush opened 4 months ago

srush commented 4 months ago

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 commented 4 months ago

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.