Commit graph

46 commits

Author SHA1 Message Date
lucasdelimanogueira
240e603c0c fix distributed dataloader 2024-05-27 15:09:07 -03:00
lucasdelimanogueira
e3bb882d3c fix distributed dataloader 2024-05-27 15:08:27 -03:00
lucasdelimanogueira
10a88fe8ca unit test dataloader and distributedsampler 2024-05-25 11:55:59 -03:00
lucasdelimanogueira
4676a70d14 fix max min sum access elements autograd backward running on device 2024-05-23 12:16:07 -03:00
lucasdelimanogueira
f13d22eb91 fix send to device unittest 2024-05-23 10:04:44 -03:00
lucasdelimanogueira
b1c28c4cea fix send to device 2024-05-23 10:00:48 -03:00
lucasdelimanogueira
065e913b2d fix send to device 2024-05-23 09:46:35 -03:00
lucasdelimanogueira
a072b57b50 fix module send to cuda 2024-05-23 03:19:49 -03:00
lucasdelimanogueira
5946e6e1bd fix unittest test nn cuda 2024-05-23 03:01:27 -03:00
lucasdelimanogueira
d70e398773 fix autograd cuda unittests 2024-05-23 02:57:57 -03:00
lucasdelimanogueira
702ebaf87c 1D, 2D and 3D transpose cuda 2024-05-23 02:09:32 -03:00
lucasdelimanogueira
65dc86aaa4 broadcasted batched matmul cuda 2024-05-22 19:20:54 -03:00
lucasdelimanogueira
cebf890165 sigmoid kernel avoid overflow 2024-05-22 13:24:01 -03:00
lucasdelimanogueira
ed05e57cf4 squeeze operation and autograd 2024-05-21 14:43:16 -03:00
lucasdelimanogueira
be4eef45ac unittest unsqueeze and unsqueeze neg 2024-05-21 14:32:14 -03:00
lucasdelimanogueira
db434ec197 unsqueeze operation 2024-05-21 14:21:35 -03:00
lucasdelimanogueira
9cfb1f622d softmax autograd 2024-05-21 13:52:05 -03:00
lucasdelimanogueira
d530f79a82 cross entropy loss 2024-05-21 12:16:12 -03:00
lucasdelimanogueira
a57298dfcb softmax and small fixes 2024-05-21 02:03:24 -03:00
lucasdelimanogueira
a39bd59a3f softmax operation 2024-05-21 01:22:38 -03:00
lucasdelimanogueira
fd831f950d fix add min max keepdim 2024-05-21 00:38:34 -03:00
lucasdelimanogueira
5e72fe57d3 softmax v1 need fix some erros yet 2024-05-20 20:16:41 -03:00
lucasdelimanogueira
cd57c39ff0 min max broadcasted autograd and small fix on add broadcasted 2024-05-20 18:44:43 -03:00
lucasdelimanogueira
b324b7e1fa broadcasted equal operation 2024-05-20 16:42:45 -03:00
lucasdelimanogueira
098ede5a6b min max autograd 2024-05-20 15:24:55 -03:00
lucasdelimanogueira
cc483417dc max min and equal operations 2024-05-20 12:50:45 -03:00
lucasdelimanogueira
5dd3f3d202 implementing keepdim sum 2024-05-17 19:08:09 -03:00
lucasdelimanogueira
abcd17e679 fix test autograd sub broadcasted 2024-05-17 13:49:32 -03:00
lucasdelimanogueira
de72cd67ba sub broadcasted cuda 2024-05-17 13:15:33 -03:00
lucasdelimanogueira
64fb3f9edb fix sum axis autograd and reversed add broadcasted order 2024-05-17 11:39:26 -03:00
lucasdelimanogueira
3627e42469 sum axis autograd 2024-05-16 21:05:30 -03:00
lucasdelimanogueira
156ec9e3e3 add new test matmul cases broadcasted and batched 2024-05-16 20:34:06 -03:00
lucasdelimanogueira
c11cfd7f22 fix transpose cuda 2024-05-16 19:10:22 -03:00
lucasdelimanogueira
85d8e74ad1 fix cuda add broadcasted 2024-05-16 18:06:50 -03:00
lucasdelimanogueira
ddf04363d1 sub broadcasted cpu autograd 2024-05-16 17:03:09 -03:00
lucasdelimanogueira
27a866d721 add broadcasting autograd 2024-05-16 16:43:11 -03:00
lucasdelimanogueira
3c8da6a245 sum axis cpu and broadcast forward 2024-05-16 16:35:55 -03:00
lucasdelimanogueira
7c7092eb8d small fix unittests 2024-05-10 21:01:25 -03:00
lucasdelimanogueira
e6104e43c9 small changes 2024-05-10 20:12:15 -03:00
lucasdelimanogueira
6b08c10300 Cosine and sine operations with autograd 2024-05-09 22:32:36 -03:00
lucasdelimanogueira
7e3100f796 unit test nn module mse loss and sigmoid 2024-05-07 17:44:59 -03:00
lucasdelimanogueira
d767fb5e2a unit test cuda 2024-05-07 16:42:00 -03:00
lucasdelimanogueira
01b62a8c71 unit test autograd transpose reshape T 2024-05-07 16:29:32 -03:00
lucasdelimanogueira
c5a776f43d unit test autograd transpose, t and reshape then matmul 2024-05-07 13:39:18 -03:00
lucasdelimanogueira
708fc121d6 unit test autograd 2024-05-07 12:48:06 -03:00
lucasdelimanogueira
2c2569d482 operations unit tests 2024-05-07 11:52:56 -03:00