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

1501 Commits

Author SHA1 Message Date
Nathaniel Simard 8ed0775dfd
Perf/wgpu/context (#395) 2023-06-12 14:24:19 -04:00
Louis Fortier-Dubois fd88398ce4
feat/Xavier glorot (#394) 2023-06-11 09:54:44 -04:00
Mathias Insley f8cd38c071
Feat/gru (#393) 2023-06-09 18:56:40 -04:00
Nathaniel Simard 8c9802c363
Feat/wgpu/reduction (#392) 2023-06-08 16:54:36 -04:00
uint256_t dc5aa41ea5
ONNX: Add `Bool` element type (#389) 2023-06-08 16:48:51 -04:00
Nathaniel Simard cb4f049eae
fix: improve assert approx (#390) 2023-06-06 14:34:20 -04:00
Mathias Insley 8a88a868ee
Feat/lstm (#370) 2023-06-06 14:33:22 -04:00
Nathaniel Simard bff752b1a8
Feat/wgpu/index (#386) 2023-06-06 12:29:14 -04:00
Nathaniel Simard c1e1e38a79
Add more ops (#387) 2023-06-06 12:21:20 -04:00
Nathaniel Simard ecc67c58f9
Feat/wgpu/swap dims (#381) 2023-06-04 19:34:35 -04:00
Yu Sun 105c259d44
feat(learner): add RegressionOutput (#380) 2023-06-04 10:21:29 -04:00
Nathaniel Simard 0a205a3603
feat: add basic matmul for wgpu backend (#379) 2023-06-03 10:32:57 -04:00
Yu Sun 498d163a7b
Feat: mse loss (#378) 2023-06-03 10:31:12 -04:00
Nathaniel Simard 974fdfaba1
Feat/wgpu backend setup (#376) 2023-06-02 11:52:47 -04:00
Dilshod Tadjibaev 483f9acca5
Enhance SqliteDatasetWriter to remove tmp on SIGTERM (#375) 
Temporary files were dangling when a process is terminated by SIGINT (ctrl/cmd+c) or SIGTERM. This change used gix-tempfile to wrap a temporary file with a special signal handle to will remove the tmp file when writer is dropped or when program is terminated early.
 2023-06-02 09:49:48 -04:00
Dilshod Tadjibaev f24f91c651
Fix clippy error (#377) 2023-06-01 17:11:20 -04:00
Nathaniel Simard 3b6ba1bbab
Refactor/dim inference (#369) 2023-06-01 12:02:12 -04:00
Dilshod Tadjibaev f05cf7bac3
Add SqliteDatasetWriter to store items in sqlite db (#364) 2023-06-01 09:54:50 -04:00
Dilshod Tadjibaev befec6bcb2
Dataset audio: add Copy derive (#372) 2023-05-31 10:57:30 -04:00
Nathaniel Simard bda5c237b0
refactor: onnx parsing (#368) 2023-05-27 18:44:57 -04:00
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