chalk icon indicating copy to clipboard operation
chalk copied to clipboard

Published 20 hours ago • verified publisher icon chalk-diagrams

Support numpy/jax

Open srush opened this issue 2 months ago • 1 comments

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.

srush avatar May 14 '24 21:05 srush