5 Commits

Author	SHA1	Message	Date
Francois Chollet	89b2ea5977	Add fully operational JAX trainer.	2023-04-18 14:49:38 -07:00
Francois Chollet	f4108a9e9d	Add main loss tracking in TF trainer	2023-04-17 17:23:53 -07:00
Francois Chollet	e363fb2b2d	Misc fixes.	2023-04-17 15:41:48 -07:00
Francois Chollet	8fa442ccbe	Progress on compile utils	2023-04-16 12:21:29 -07:00
Francois Chollet	160473c474	Initial draft of the codebase.	2023-04-09 12:21:45 -07:00