4 Commits

Author	SHA1	Message	Date
Ahmed Harmouche	2114dc13d1	Allow multi-input model export (#1995) ... * Allow multi-input model export * Add model export unit test * Fix efficientnet compilation * Only run model export test on JIT supported devices * Skip export model test if not EXPORT_SUPPORTED_DEVICE	2023-10-07 04:13:34 -07:00
George Hotz	718ced296c	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
chenyu	ae39cf84ab	Symbolic Shape JIT main PR (#1353) ... * Symbolic Shape JIT update tests 2 variables symbolic ops, adding more tests test passing cleanup * more test cases * single flag * review update * jit attention one piece * realize * symbolic_jit test for cuda * old artifact * works with cuda gpu but failed ci * CUDACPU	2023-08-18 14:39:55 -07:00
Diogo	4dc8595069	simple exporting models (#1344) ... * unified exporting * json exporting * ignore more * simplified buffer export * added dtypes * added assert * swift example * fix tests * linter * remove whitespace * fixed tests * remove swift example * remove unintended changes * allow callable models to be used * whitespace * more readable json export * name change * whitespace * whitespace	2023-08-01 09:35:48 -07:00