Figma now has AI motion graphics and shader tools (opens in new tab)

You can also work directly with code on the Figma Design canvas, thanks to code layers.