lucasdelimanogueira
|
6701179c22
|
del tensor free memory
|
2024-06-05 14:31:48 -03:00 |
|
lucasdelimanogueira
|
e590f57555
|
implemented distributed sampler
|
2024-05-25 10:53:09 -03:00 |
|
lucasdelimanogueira
|
b152ae2a1d
|
first commit distributed module
|
2024-05-23 19:15:30 -03:00 |
|
lucasdelimanogueira
|
65dc86aaa4
|
broadcasted batched matmul cuda
|
2024-05-22 19:20:54 -03:00 |
|
lucasdelimanogueira
|
40a6bccab4
|
vision tranform, readme and example training
|
2024-05-22 14:25:02 -03:00 |
|
lucasdelimanogueira
|
c808a49a30
|
dataset, dataloader and mnist
|
2024-05-18 17:59:14 -03:00 |
|
lucasdelimanogueira
|
a846d3aa3e
|
Small fix for PiPy
|
2024-05-11 01:45:59 -03:00 |
|
lucasdelimanogueira
|
7e3100f796
|
unit test nn module mse loss and sigmoid
|
2024-05-07 17:44:59 -03:00 |
|
lucasdelimanogueira
|
e5819a4121
|
nn module v1
|
2024-05-06 19:47:27 -03:00 |
|
lucasdelimanogueira
|
8366245813
|
sub, elementmul, pow cuda operations
|
2024-04-30 00:31:46 -03:00 |
|