6 Commits

Author	SHA1	Message	Date
Awni Hannun	99abb9eff4	Async eval (#972)	2024-04-09 18:34:00 -07:00
Awni Hannun	1b97b2958b	Compile with capture (#629) ... * Simple kernel generation * Remove the generate kernel from graph_utils * fix multi-output with compile * fuse with stopgrad * v1 input, output capture in compile * cleanup tree update with visitor update * nit * remove todo * state for model, optional explicit init and more pure optimizer steps * move learning rate to state * add lr to opt state, some fixes in capture * fix optim * update tuple of containers as well * fix stream for compiled output * rng state for compile * nit * updates and comments --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-02-07 17:29:22 -08:00
Angelos Katharopoulos	a611b0bc82	Removes the retain_graph flag (#385) ... * Adds global tracing flag * Removes retain_graph in favor of is_tracer	2024-01-07 15:16:51 -08:00
Zach Schillaci	5b9be57ac3	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00