[Google Research] Improving brain models with ZAPBench