Zhuohan Li, UC Berkeley