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

331 Commits

Author SHA1 Message Date
Nathaniel Simard 2a4ba5a6ab
Feat/gather scatter (#367) 2023-05-27 11:40:04 -04:00
Dilshod Tadjibaev 41e54f741b
Use sqlite bundled version (#366) 
Fixes #365
 2023-05-26 18:09:01 -04:00
Nathaniel Simard 3bd58d87ef
feat: burn import logging (#363) 2023-05-25 17:54:51 -04:00
Nathaniel Simard 3ef2a18d87
Fix flaky tests + add feature flag (#362) 2023-05-24 08:33:31 -04:00
Dilshod Tadjibaev 56e40ae63b
Dataset audio - Speech Commands (#361) 2023-05-22 14:24:02 -04:00
Nathaniel Simard ba0ac112e1
Refactor: Burn Import use BurnGraph (#355) 2023-05-21 13:09:27 -04:00
Dilshod Tadjibaev af31d018ee
Fix test error due to an updated model serializer (#357) 2023-05-21 09:08:30 -04:00
Dilshod Tadjibaev f5f38d3f78
Update sqlite version (#358) 
Fixes #356.
 2023-05-21 09:08:06 -04:00
Dilshod Tadjibaev b170b539b7
Upgrade dep versions (#359) 2023-05-21 09:07:39 -04:00
Dilshod Tadjibaev 6fece7e4cb
Dataset Improvements: Add Sqlite storage backend and HF importer improvements (#353) 2023-05-20 14:24:55 -04:00
Nathaniel Simard 18cb19cd03
Fix deserializer array of constant (#354) 2023-05-19 12:01:59 -04:00
Nathaniel Simard 510d53f37e
Improve softmax doc (#352) 2023-05-16 14:39:42 -04:00
Nathaniel Simard 2df268b674
feat: improve module recorder deserialization (#351) 2023-05-15 08:29:57 -04:00
Nathaniel Simard 976102fec0
Feat/avg pool1d (#349) 2023-05-15 08:29:45 -04:00
Dilshod Tadjibaev 43154ce7e0
Move dataset.py into its own file and add CustomEncoder (#348) 2023-05-12 09:38:09 -04:00
Dilshod Tadjibaev fbbc8ac560
Remove build panic from burn-tch (#347) 2023-05-12 09:37:58 -04:00
Nathaniel Simard 8bbfbd8a8f
perf: softmax (#346) 2023-05-12 09:37:42 -04:00
Nathaniel Simard 747e245cc4
chore: bump tch version (#345) 2023-05-11 14:21:52 -04:00
Dilshod Tadjibaev 05763e1878
Bump version to the next minor to indicate dev (#344) 2023-05-10 18:02:08 -04:00
Nathaniel Simard 73f99ef79f
Feat/maxmin numeric (#340) 2023-05-09 16:35:55 -04:00
Nathaniel Simard a88357ce1d
feat: add max & min ops (#339) 2023-05-09 15:49:33 -04:00
Nathaniel Simard 69001b0d69
Feat/activation ops (#338) 
* perf: GELU

* Refactor relu
 2023-05-09 08:32:35 -04:00
nathaniel 844b199dc1 Add burn import description 2023-05-06 10:45:58 -04:00
Nathaniel Simard 29eecd6383
Prepare next release (#335) 2023-05-06 10:32:23 -04:00
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