tensor-canvas (pypi): A 2D graphics library that draws directly onto tensors and is cross-framework compatible
Keywords: 2d|graphics|jax|ml|pytorch|sdf|tensorflow