Commit Graph

264 Commits

Author SHA1 Message Date
Nathaniel Simard f8ab29bba6
feat: add comparison (#19) 2022-08-22 14:27:23 -04:00
Nathaniel Simard 5560bae26a
Feat/elem conversion (#18) 2022-08-22 13:27:15 -04:00
Nathaniel Simard 0f6b50fc10
Feat/dataloader (#17) 2022-08-22 11:21:22 -04:00
Nathaniel Simard 1b8b1e3843
Feat/dataset huggignface (#16) 2022-08-21 17:20:45 -04:00
nathaniel 6ef4653d21 feat: add burn-dataset 2022-08-21 09:42:42 -04:00
nathaniel 39be42b2c6 feat: basic setup 2022-08-20 20:20:56 -04:00
nathaniel 0f38096abe feat: cross entropy loss 2022-08-20 17:27:04 -04:00
Nathaniel Simard 6642a49d84
Feat/softmax (#15) 2022-08-20 17:13:55 -04:00
nathaniel 7e78aa09ea Feat: add div ops 2022-08-20 12:15:58 -04:00
Nathaniel Simard b2912f483a
Feat/more ops (#14) 2022-08-20 11:20:25 -04:00
nathaniel 71d37f1ec6 feat: relu activation function 2022-08-10 19:09:43 -04:00
nathaniel adf56ed9e6 refactor: fix tensor generics order 2022-08-09 12:40:51 -04:00
nathaniel 77d7c84ab3 Fix element 2022-08-08 19:18:54 -04:00
nathaniel 127b7cb60b wip 2022-08-08 19:18:54 -04:00
nathaniel 5d6c866198 feat: implement mask fill ops 2022-08-08 08:23:55 -04:00
Nathaniel Simard c261398814
Feat/more ops (#11) 2022-08-07 17:28:08 -04:00
Nathaniel Simard 0059c611f5
Feat/module definition (#10) 2022-08-05 21:14:44 -04:00
nathaniel e2f3a6fa10 feat: add backend infos 2022-07-31 14:01:42 -04:00
Nathaniel Simard d00b74b527
Feat/tensor concrete type (#9) 2022-07-31 12:06:25 -04:00
nathaniel 2872b488ef feat: remove some where for ndarray 2022-07-28 14:54:03 -04:00
nathaniel 5f9e0d1bac ci: publish to crates.io 2022-07-27 16:54:34 -04:00
nathaniel bfbdfbeb77 doc: update README 2022-07-27 16:15:48 -04:00
nathaniel 305438a476 Merge branch 'main' of github.com:nathanielsimard/burn 2022-07-27 15:45:10 -04:00
nathaniel ae21cfa34d doc: add burn-tensor Readme 2022-07-27 15:44:41 -04:00
Nathaniel Simard 43f2d58a82
Update README.md 2022-07-27 14:37:48 -04:00
nathaniel 746cf79c4b ci: setup ci 2022-07-27 12:26:54 -04:00
nathaniel 7698dd7684 refactor: cleanup backends 2022-07-27 11:56:30 -04:00
nathaniel fbf5fe3dda refactor: move zeros + ones impl 2022-07-27 11:42:50 -04:00
nathaniel acbbcc3781 feat: add creation trait 2022-07-27 11:24:41 -04:00
nathaniel b0f0d7e378 refactor: create Tensor + Element trait 2022-07-27 10:20:21 -04:00
nathaniel 122cd842a2 refactor: move tensor base ops 2022-07-27 10:05:38 -04:00
nathaniel 6f45e878f1 refactor: move zeros + ones trait in tensor ops 2022-07-27 09:43:57 -04:00
nathaniel 51cb331aef refactor: imports 2022-07-27 09:37:56 -04:00
nathaniel 66cd4ccb9c perf: optimize from bmatrix for ndarray 2022-07-26 18:12:26 -04:00
nathaniel 3ba6c69875 feat: full support for ndarray 2022-07-26 18:01:29 -04:00
nathaniel 7dcd9d5507 feat: wip add ndarray 2022-07-26 15:51:11 -04:00
nathaniel ec32fa730c feat: implement more ops for ndarray backend 2022-07-26 14:21:07 -04:00
nathaniel b2f3c42376 feat: wip ndarray backend 2022-07-26 12:22:24 -04:00
nathaniel c7dad3977b feat: support higher order autodiff 2022-07-26 10:14:17 -04:00
nathaniel 61b67b44ff doc: add a simple example 2022-07-26 10:04:13 -04:00
nathaniel b306156cc2 feat: init tensor 2022-07-26 09:45:13 -04:00
nathaniel fe1b9b6972 feat: support transpose un auto diff 2022-07-26 07:59:24 -04:00
nathaniel 9af7bde608 feat: support neg in auto diff 2022-07-26 07:52:24 -04:00
nathaniel 445a9fbcbe feat: support neg in auto diff 2022-07-26 07:52:17 -04:00
nathaniel 824e1a345f feat: support index + index_assign with auto diff 2022-07-25 22:48:37 -04:00
nathaniel f939b5c775 feat: support reshape in AD 2022-07-25 20:55:13 -04:00
nathaniel ae994f367a feat: support half precision 2022-07-25 19:52:40 -04:00
nathaniel 0dac3f6bdf refactor: ad element 2022-07-25 19:40:14 -04:00
Nathaniel Simard 3769b63305
refactor: breadth first search graph traversal (#5) 2022-07-25 19:10:48 -04:00
Nathaniel Simard a226eabcc8
Refactor/stateless forward node (#4)
* refactor: forward + backward graph

* refactor: always use execute_ops for all ops

* refactor: extract forward and backward nodes

* feat: forward multithread safe

* refactor: backward state

* feat: backward multi-threaded

* feat: remove multi-thread backward
2022-07-25 17:59:25 -04:00