7 Commits

Author	SHA1	Message	Date
George Hotz	bd9c015b09	tests from grad uop path [pr] (#8313)	2024-12-18 09:25:05 -08:00
George Hotz	bcd7ea60f0	hotfix: a few more grad tests	2024-12-13 21:03:02 -08:00
George Hotz	734f2c5344	compute gradient [pr] (#8237) ... * compute gradient [pr] * schedule_step_with_grads * second deriv works	2024-12-13 20:46:01 -08:00
George Hotz	8396d90f91	non controversial changes from optim branch [pr] (#8234)	2024-12-13 19:24:16 -08:00
George Hotz	37fa38d272	Revert "switch beautiful_mnist to use new optimizer [pr] (#8231)" (#8233) ... This reverts commit e9ee39df22.	2024-12-13 19:07:09 -08:00
George Hotz	e9ee39df22	switch beautiful_mnist to use new optimizer [pr] (#8231) ... * switch beautiful_mnist to use new optimizer [pr] * fix abstractions3 + docs * fix OptimizerGroup with schedule_step api	2024-12-13 18:27:16 -08:00
George Hotz	e2f87ecf36	start work on new gradient (#7838) ... * start work on new gradient * more correct * working tests * more tests * work * add (faliing) gradient test * add view and reduce gradient * test_add works, many failing test_ops * add max and reduce max * add max and reduce max * 129 failing * 108 failed * better view drawing * 101 failed * i got 99 failures * 94 failures * it's tons of terrible code, but only 50 tests fail * only 19 failures * same 19 but shorter * minimal doesn't matter * shorter * lil simpler * simpler * simpler * simpler * 13 test failures * nine tests fail * all ops tests pass * add contiguous gradient + fix sched tests * faster by removing toposort calls * missed one * add jax to testing	2024-12-13 16:45:53 -08:00