Diagram

Diagram builds AI tools with a focus on design. They offer tools that you can combine with Figma which will boost your workflow a lot.