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

507 Commits

Author SHA1 Message Date
Dilshod Tadjibaev b7b4f321eb
Correct the struct name for ModelGen (#328) 2023-05-05 12:57:02 -04:00
Nathaniel Simard a50262b730
refactor: grad clipping (#326) 2023-05-05 12:56:35 -04:00
Nathaniel Simard be72aa09ab
doc: update record architecture doc (#329) 2023-05-05 12:53:19 -04:00
Mathias Insley d892f6186c
Gradient Clipping (#305) 2023-05-04 15:01:30 -04:00
Nathaniel Simard 6c9a5c8e58
Refactor/record (#323) 2023-05-04 14:59:16 -04:00
Dilshod Tadjibaev 39297a6479
Readme updates (#325) 
* Update text-generation readme for Mac users

* Update root readme to reference import crate

* Update import's readme

* Update torch backend
 2023-05-04 14:58:44 -04:00
Dilshod Tadjibaev 314db93b7f
Add ability to load onnx state to the generated source code (#319) 2023-05-03 13:05:43 -04:00
Nathaniel Simard b54f9302c7
Feat/avg pool2d (#318) 2023-04-30 12:25:14 -04:00
Nathaniel Simard 44ffaa3ad2
Add transformer decoder (#316) 2023-04-27 10:00:30 -04:00
Nathaniel Simard a835b48140
Fix: support broadcast during backward (#312) 2023-04-26 09:20:27 -04:00
Nathaniel Simard 0d03a2752e
Feat/improve serialization settings (#311) 2023-04-25 12:46:50 -04:00
Nathaniel Simard 4773b7fb42
doc: link architecture doc (#310) 2023-04-25 12:46:29 -04:00
Nathaniel Simard 2865000080
doc: add tensor + backend architecture doc (#309) 2023-04-25 09:24:38 -04:00
Nathaniel Simard c9fda41846
Doc: initial architecture documentation (#308) 2023-04-24 13:07:30 -04:00
Sunny Gonnabathula 02abc373d3
add burn-tch support for bf16 (#303) 2023-04-24 11:24:36 -04:00
Nathaniel Simard c5e31b272f
Feat/group conv (#306) 2023-04-22 15:00:41 -04:00
Nathaniel Simard 78ac09fb7a
Support dilation in convolution operations (#301) 2023-04-18 10:01:11 -04:00
Nathaniel Simard bd58922784
Feat/conv stride (#300) 2023-04-16 22:03:16 -04:00
Dilshod Tadjibaev df980d534e
Add foundation for importing ONNX files (#297) 2023-04-15 10:44:50 -04:00
Dilshod Tadjibaev a74f26620a
Create POEM.md (#299) 
Adding the soul for the project: a poem by ChatGPT.
 2023-04-13 09:00:39 -04:00
Dilshod Tadjibaev 8350aedac8
Move publish.sh under scripts directory (#298) 2023-04-13 08:59:23 -04:00
Yu Sun 26cf555612
Feat: add new tensor ops mask_scatter (#258) 2023-04-13 08:58:31 -04:00
Dilshod Tadjibaev f872a1aa21
Update landing README.md (#296) 
I made a few things more prominent and updated wording, so that it's more understandable what the project is about.
 2023-04-12 15:29:59 -04:00
Sunny Gonnabathula 69954c14ec
add bf16 element (#295) 2023-04-12 12:36:03 -04:00
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 4a0312644f
Fix test template (#291) 2023-04-11 15:43:22 -04:00
Dilshod Tadjibaev 0a288dcb79
Add run-doc-test flat in test CI template (#290) 2023-04-11 13:59:26 -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 1c9ae2b5ad
Create FUNDING.yml 2023-04-10 19:33:02 -04:00
Nathaniel Simard 2220965b5c
Feat/add tensor checks (#283) 2023-04-10 19:16:15 -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 d617c2cf69 Fix CI 2023-04-08 11:15:47 -04:00
Dilshod Tadjibaev b8283b0dbf
Change rust action to fix #263 (#277) 2023-04-08 10:53:08 -04:00
Nathaniel Simard f04fe101d8
Feat/module no grad (#274) 2023-04-07 09:01:27 -04:00
Mathias Insley d8f64ce1dd
Pretty Print Tensors (#257) 2023-04-06 20:06:38 -04:00
Nathaniel Simard ca8ee0724d
Refactor/optim (#272) 2023-04-05 12:38:53 -04:00
Nathaniel Simard 0f6172ffc6
Create record derive (#271) 2023-04-04 15:16:20 -04:00
Nathaniel Simard b2cf37eb8b
Fix: load checkpoints (#270) 2023-04-04 13:09:58 -04:00
Dilshod Tadjibaev 7e0f127c44
Add Code of Conduct (#269) 2023-04-03 18:32:20 -04:00
Nathaniel Simard 2c151a5570
Update module doc + add sponsors section (#267) 2023-04-02 17:37:01 -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
Dilshod Tadjibaev 4e9e6d2706
Move unsqueeze op to the tensor's base (#261) 2023-04-01 14:20:48 -04:00
Nathaniel Simard 32d38bebc3
Refactor Param wrapping only for Tensor (#259) 2023-03-31 16:45:10 -04:00
Dilshod Tadjibaev 7364d09d32
Add flatten op to the tensor base (#260) 2023-03-31 16:44:48 -04:00
Nathaniel Simard c14c7977ec
Text classification inference example (#256) 2023-03-26 16:51:37 -04:00