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
..
module-base-class.rst	Module checks the weight on load_weights (#337)	2024-01-02 18:55:42 -08:00
optimizers-template.rst	awni's commit files	2023-11-29 10:30:41 -08:00