Shaojie Bai

Member of Technical Staff at Thinking Machines Lab

Bai co-created deep equilibrium models, which repeatedly refine one internal state until it settles on an answer instead of storing the output of many separate layers. The approach can match much larger conventional networks while using substantially less memory.