dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,501 Commits 39 Branches 19 Tags 101 MiB
Commit Graph

263 Commits

Author SHA1 Message Date
Nathaniel Simard c9e344a97f
Fix/batch norm (#238) 2023-03-17 09:31:52 -04:00
Dilshod Tadjibaev ed5888e701
Mnist inference web: Readme (#233) 2023-03-15 08:49:59 -04:00
Nathaniel Simard a314c7b694
Perf/optimize backward ops (#232) 2023-03-14 12:59:52 -04:00
Dilshod Tadjibaev e991ba38df
Mnist inference web: Remove wee_alloc allocator and use rust's alloc for WASM32 (#231) 2023-03-14 08:57:10 -04:00
nathaniel 7f3f8d2b22 fix: mnist inference web build for web no such dir 2023-03-13 20:13:39 -04:00
Dilshod Tadjibaev bf9d33e6fc
Add MNIST inference on the web demo crate (#228) 
* Add MNIST inference on the web demo crate

* Fix problems identified during a PR review
 2023-03-13 19:51:32 -04:00
Nathaniel Simard 453f4aec86
feat: pre-norm transformer (#230) 2023-03-13 19:09:35 -04:00
Nathaniel Simard 403500f018
feat(burn-core): refactor cross entropy (#229) 2023-03-13 13:47:19 -04:00
Nathaniel Simard be96160065
refactor: elems (#206) 2023-03-06 18:39:49 -05:00
Nathaniel Simard 019c5f9c44
Refactor/int backend (#197) 
* Update burn-tensor API

* Migrate burn-autodiff

* Update burn-tch

* Update burn-ndarray

* Add some doc
 2023-03-06 14:45:58 -05:00
Nathaniel Simard ffd3d35176
Refactor/tensor api (#191) 2023-03-05 09:23:42 -05:00
Nathaniel Simard 25deb5a13b
Refactor/autodiff (#186) 2023-02-28 20:01:26 -05:00
Dilshod Tadjibaev c510e8bfb5
Add accelerate framework support for burn-ndarray (#183) (#184) 2023-02-28 09:09:47 -05:00
Dilshod Tadjibaev fb925acc73
Make burn and burn-core packages no_std compatible (#168) (#173) 
* Make burn-ndarray and burn-tensor no_std compatible (#168)
 2023-02-25 09:38:01 -05:00
Nathaniel Simard e4e9a5dcca
fix(burn-tch): tensor comparison ops inplace don't have the right tch kind (#166) 2023-02-18 13:59:49 -05:00
Nathaniel Simard 7d2f43dfca
Refactor Tensor API (#163) 2023-02-17 17:31:20 -05:00
Nathaniel Simard c7963d8485
refactor: device functions (#157) 2023-01-27 18:37:21 -05:00
Nathaniel Simard b7aa066e5c
Feat/nn max pool2d (#153) 2023-01-21 17:06:39 -05:00
Nathaniel Simard eea5a263bf
Feat/adam optimizer (#140) 2022-12-30 15:02:43 -05:00
Visual 567adfb93e
refactor: fix all clippy warnings (#137) 2022-12-25 11:22:25 -05:00
Visual 85f98b9d54
refactor, feat: clean Cargo.toml files, upgrade tch to 0.10 (#131) 
* Clean Cargo.toml files, upgrade tch to 0.10

* Add pull_request hook to test.yml workflow
 2022-12-25 10:36:23 -05:00
Nathaniel Simard 3a9dfe6097
feat: cross entropy loss (#130) 2022-12-25 10:10:22 -05:00
Nathaniel Simard 3a91c2c48e
Feat/multi device (#128) 2022-12-20 18:01:58 -05:00
Nathaniel Simard a599eaed88
Feat/module visitor (#127) 2022-12-17 14:25:36 -05:00
Nathaniel Simard d9592411c2
Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00
Nathaniel Simard eee90a5c9e
Example/text classification (#123) 2022-12-02 17:42:49 -05:00
Nathaniel Simard 46d06f0c90
feat: module init (#117) 2022-11-25 22:02:26 -05:00
Nathaniel Simard 4abc281dae
Refactor/forward pass (#116) 2022-11-25 21:14:44 -05:00
Nathaniel Simard acb14adc29
Fix/named tensor (#115) 2022-11-25 19:30:35 -05:00
Nathaniel Simard e0e787f87d
Experimental/named tensor (#113) 2022-11-23 19:05:46 -05:00
Nathaniel Simard c4c739d91b
doc: update readme (#112) 2022-11-20 19:41:55 -05:00
Nathaniel Simard dda067e79b
Refactor/backend autodiff (#106) 2022-11-19 12:37:06 -05:00
Nathaniel Simard d45d674a04
Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
Nathaniel Simard ab51c22a55
Refactor/extract tch backend (#103) 2022-11-15 21:06:40 -05:00
Nathaniel Simard 3da122db09
Add cuda gpu example + doc (#91) 2022-11-09 21:32:51 -05:00
Nathan Henrie 947ed00301
Fix `cargo run --example mnist` (#90) 2022-11-09 20:45:58 -05:00
Nathaniel Simard c5213b6c32
Fix: mnist example (#85) 2022-11-07 17:58:50 -05:00
Nathaniel Simard 5b4855317b
Doc/improve example (#64) 2022-10-26 18:08:38 -04:00
Nathaniel Simard bd36023b06
Update projects (#29) 2022-09-04 14:22:56 -04:00
Nathaniel Simard 548a1be488
Refactor/burn tensor (#28) 2022-09-03 12:51:38 -04:00
Nathaniel Simard 1ff11e537a
Feat/save load (#27) 2022-09-01 20:31:45 -04:00
Nathaniel Simard 674e078a85
Feat/inner module (#26) 2022-08-30 18:05:42 -04:00
nathaniel 68e052513b refactor: trainer 2022-08-27 11:54:51 -04:00
nathaniel 2c4288d536 fix integer backend 2022-08-27 11:34:53 -04:00
nathaniel 9ed0b0bdcd refactor: multi-thread logger -> async logger 2022-08-26 20:59:21 -04:00
nathaniel 05eb2f9cfd feat: async logging 2022-08-26 20:55:42 -04:00
nathaniel 32d81e4fdd refactor: dataloader 2022-08-26 20:25:19 -04:00
nathaniel 9ed252755c chore: cleanup 2022-08-26 18:38:16 -04:00
Nathaniel Simard 8da87b376d
Feat/use index assign for cat (#24) 2022-08-26 15:41:29 -04:00
Nathaniel Simard 508cfd2b5a
Refactor/metric (#23) 2022-08-23 16:39:30 -04:00
Nathaniel Simard d62f2b09ad
Feat/argmax (#22) 2022-08-23 13:57:30 -04:00
Nathaniel Simard 0aa486fc0c
Feat/logger (#21) 2022-08-23 11:47:21 -04:00
nathaniel bad608e72c feat: supervised trainer 2022-08-22 19:54:33 -04:00
Nathaniel Simard 7675ca0776
Feat/mix precision (#20) 2022-08-22 16:03:53 -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 39be42b2c6 feat: basic setup 2022-08-20 20:20:56 -04:00
Nathaniel Simard d00b74b527
Feat/tensor concrete type (#9) 2022-07-31 12:06:25 -04:00
nathaniel b0f0d7e378 refactor: create Tensor + Element trait 2022-07-27 10:20:21 -04:00
nathaniel 51cb331aef refactor: imports 2022-07-27 09:37:56 -04:00
nathaniel 3ba6c69875 feat: full support for ndarray 2022-07-26 18:01:29 -04:00
nathaniel 61b67b44ff doc: add a simple example 2022-07-26 10:04:13 -04:00