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

74 Commits

Author SHA1 Message Date
Nathaniel Simard 66028dc3cf
Feat/lr scheduler (#276) 2023-04-08 13:12: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 7d7504686a
Refactor/init modules (#250) 2023-03-24 15:38:02 -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 Simard a314c7b694
Perf/optimize backward ops (#232) 2023-03-14 12:59:52 -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 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 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 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