7 Commits

Author	SHA1	Message	Date
Francois Chollet	df6990f41a	Misc fixes + add support for JAX array input	2023-04-17 17:47:22 -07:00
Francois Chollet	e363fb2b2d	Misc fixes.	2023-04-17 15:41:48 -07:00
Francois Chollet	9e4c590ac4	Misc fixes	2023-04-17 14:55:17 -07:00
Francois Chollet	90df7ef9fc	Format codebase	2023-04-16 13:30:21 -07:00
Francois Chollet	b851873765	Refine correctness of CompileMetrics	2023-04-16 13:28:02 -07:00
Francois Chollet	8fa442ccbe	Progress on compile utils	2023-04-16 12:21:29 -07:00
Francois Chollet	8c199becab	Further work on Trainers module	2023-04-15 18:51:10 -07:00