Figma's implementation uses WebGL so it might be difficult to pull off. I think there's some extra resources needed to act like flexbox, compared to actual flex box.