Commit Graph

206 Commits

Author SHA1 Message Date
0xNaN adbfc67456 test `jacobian` and `numerical_jacobian` against torch.autograd.functional.jacobian 2020-10-22 01:28:52 +02:00
0xNaN 7458089b3b fix shape of Jacobian to (n_out, n_in) 2020-10-22 01:28:52 +02:00
0xNaN 1561d3b9c0 extracting `jacobian` and `test_jacobian` 2020-10-22 01:28:52 +02:00
0xNaN 93bc3c22a0 tiny gradcheck 2020-10-22 01:28:52 +02:00
George Hotz e8feaa53d6
Merge pull request #12 from adriangb/pytest-again
TST: Pytest round 2
2020-10-21 16:19:43 -07:00
Adrian Garcia Badaracco 9a8be135a7
incorporate changes 2020-10-21 13:21:44 -05:00
Adrian Garcia Badaracco 58b4f191a4
Merge remote-tracking branch 'upstream/master' into pytest-again 2020-10-21 13:19:58 -05:00
George Hotz 5a1ff084d8
Merge pull request #13 from geohot/numba
use numba to double conv speed
2020-10-21 09:51:42 -07:00
George Hotz 685ce5ba7e scope better and don't break tests 2020-10-21 09:44:28 -07:00
George Hotz a68ead09c0 use numba to double conv speed 2020-10-21 09:34:16 -07:00
Adrian Garcia Badaracco 02adb0ac3a
Make test_mnist runnable by pytest and directly 2020-10-21 11:30:08 -05:00
Adrian Garcia Badaracco 5afe6b1f68
rename files 2020-10-21 11:28:03 -05:00
George Hotz dc325af392 sneaky float64s 2020-10-21 09:26:01 -07:00
George Hotz ecdf2239fc todo 2020-10-21 09:14:37 -07:00
George Hotz d91902948b add reshape support and OMG the CONVS are SO SLOW 2020-10-21 09:12:19 -07:00
George Hotz 03fc6e8a49 layer init for all sizes 2020-10-21 08:48:16 -07:00
George Hotz e3110c9922 backward pass for conv2d, lol i mostly guessed and made shapes match 2020-10-21 08:45:35 -07:00
George Hotz 9172bc9c38
Merge pull request #10 from TheMrGhostman/master
fixing adam
2020-10-21 07:37:31 -07:00
George Hotz 3a111c832c
Merge pull request #9 from orena1/patch-1
easier to find what micrograd is...
2020-10-21 07:36:58 -07:00
TheMrGhostman fc6f355c6a fixing adam 2020-10-20 23:29:26 +02:00
Oren Amsalem a3839f0fef
easier to find what micrograd is...
Added pytorch link just for fun
2020-10-20 08:29:27 +03:00
George Hotz 4faf3a0aed simpler 2020-10-19 09:37:07 -07:00
George Hotz bee89a4840 one less loop 2020-10-19 09:34:55 -07:00
George Hotz 5c2ac48c11 write forward pass for convolution 2020-10-19 09:33:06 -07:00
George Hotz d5662e09e3 run both tests 2020-10-19 08:59:20 -07:00
George Hotz 4cace5f798 pytorch 2020-10-18 18:15:47 -07:00
George Hotz dba362e65e update readme 2020-10-18 16:40:42 -07:00
George Hotz 2681c79bc5 simple tests, repr not str 2020-10-18 14:55:20 -07:00
George Hotz 4019c38942 more readme 2020-10-18 14:38:20 -07:00
George Hotz cc9054e3ec refactor into utils 2020-10-18 14:36:29 -07:00
George Hotz 0c3dd12b3b i hate tabs 2020-10-18 14:33:13 -07:00
George Hotz 28100c741f update readme 2020-10-18 14:32:45 -07:00
George Hotz 4724b5e5b9
Merge pull request #4 from grekiki/patch-1
Make unit-tests run in all branches
2020-10-18 14:29:32 -07:00
George Hotz a139f34bb6 fix nll loss in example 2020-10-18 14:27:54 -07:00
grekiki f394a5edee
Make unit-tests run in all branches 2020-10-18 23:03:51 +02:00
George Hotz 28c9d31e49
Merge pull request #3 from dewpey/master
Test badge in readme
2020-10-18 14:02:33 -07:00
George Hotz 26ce2d93c3 add support for adam 2020-10-18 13:50:23 -07:00
Drew Patel caf03cf5d6 Added a little test badge 2020-10-18 15:41:51 -05:00
George Hotz 55f9db8af6
Merge pull request #2 from darnfish/master
Run Tests in CI
2020-10-18 13:33:58 -07:00
George Hotz 6532233d24 refactor better 2020-10-18 13:33:02 -07:00
William Gibson 91a352a8e2 Fix tests ((again)) 2020-10-18 21:31:55 +01:00
William Gibson d33885f496 Merge remote-tracking branch 'upstream/master' 2020-10-18 21:31:47 +01:00
George Hotz 92fd23df66 refactor into a few files 2020-10-18 13:30:25 -07:00
George Hotz 118c2eebe3 write sgd class 2020-10-18 13:27:59 -07:00
William Gibson c900b6ec36 Fix tests 2020-10-18 21:26:07 +01:00
William Gibson 02729c7f2a Add CI for tests 2020-10-18 21:26:01 +01:00
George Hotz f4e0cb5945 refactor tinygrad to be more tiny 2020-10-18 13:19:19 -07:00
George Hotz 54eafe6c12 update readme 2020-10-18 13:08:14 -07:00
George Hotz 83417d4b4c readme and dirs 2020-10-18 12:48:17 -07:00
George Hotz 19b3b85b23 readme 2020-10-18 11:27:37 -07:00