3 Commits

Author	SHA1	Message	Date
Rajat Sen	84e8c931d6	updating readme with new results	2024-06-03 18:03:38 +00:00
Rajat Sen	a4beacd023	Pinning jax version.	2024-05-21 20:02:57 +00:00
Yichen Zhou	b389495851	Update environments and dependencies. ... PiperOrigin-RevId: 632263761	2024-05-09 14:17:19 -07:00