dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
288 Commits 50 Branches 17 Tags 89 MiB
Commit Graph

83 Commits

Author SHA1 Message Date
Nathaniel Simard d4ce825725
refactor: burn dataset (#293) 2023-04-11 18:12:28 -04:00
Nathaniel Simard 04bcf9550a
Fix/text gen example (#292) 2023-04-11 17:18:45 -04:00
Dilshod Tadjibaev ac58aa5a70
Fix compilation error (fixes #288) (#289) 2023-04-11 16:33:17 -04:00
Dilshod Tadjibaev 546c9b386c
Update README.md (#286) 2023-04-11 12:47:30 -04:00
Nathaniel Simard f45e6863d9
fix: text classification example (#285) 2023-04-11 12:46:42 -04:00
Nathaniel Simard 37b79bcc40
Add init_with for nn modules (#280) 2023-04-08 13:12:45 -04:00
Nathaniel Simard 66028dc3cf
Feat/lr scheduler (#276) 2023-04-08 13:12:27 -04:00
Nathaniel Simard f04fe101d8
Feat/module no grad (#274) 2023-04-07 09:01:27 -04:00
Nathaniel Simard ca8ee0724d
Refactor/optim (#272) 2023-04-05 12:38:53 -04:00
Nathaniel Simard d3887bcd3d
Feat/module record (#265) 2023-04-02 16:22:05 -04:00
Nathaniel Simard 73f6d1916b
Feat/record (#262) 2023-04-02 10:09:29 -04:00
Nathaniel Simard 32d38bebc3
Refactor Param wrapping only for Tensor (#259) 2023-03-31 16:45:10 -04:00
Nathaniel Simard c14c7977ec
Text classification inference example (#256) 2023-03-26 16:51:37 -04:00
Nathaniel Simard 7d7504686a
Refactor/init modules (#250) 2023-03-24 15:38:02 -04:00
Nathaniel Simard 53baa2e10d
Feat/batch norm nd (#249) 2023-03-23 17:15:39 -04:00
Nathaniel Simard 6f43d983f7
State serialization/deserialization overhaul (#247) 2023-03-23 11:02:46 -04:00
Nathaniel Simard 4e28e2a776
chore: prepare release v0.6.0 (#246) 2023-03-21 09:47:37 -04:00
nathaniel 2f91a5e4e2 fix: burn mnist wasm demo link 2023-03-20 19:31:56 -04:00
Dilshod Tadjibaev 019ccbde4c
Fix and update readme docs (#244) 2023-03-20 11:51:07 -04:00
Dilshod Tadjibaev 8e74ce4bb9
Reduce MNIST parameters and update mnist-inference-web to match (#242) 2023-03-17 19:46:26 -04:00
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