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

69 Commits

Author SHA1 Message Date
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
Mathias Insley 8a88a868ee
Feat/lstm (#370) 2023-06-06 14:33:22 -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
Nathaniel Simard 2a4ba5a6ab
Feat/gather scatter (#367) 2023-05-27 11:40:04 -04:00
Nathaniel Simard 3ef2a18d87
Fix flaky tests + add feature flag (#362) 2023-05-24 08:33:31 -04:00
Nathaniel Simard ba0ac112e1
Refactor: Burn Import use BurnGraph (#355) 2023-05-21 13:09:27 -04:00
Nathaniel Simard 18cb19cd03
Fix deserializer array of constant (#354) 2023-05-19 12:01:59 -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 05763e1878
Bump version to the next minor to indicate dev (#344) 2023-05-10 18:02:08 -04:00
Nathaniel Simard 29eecd6383
Prepare next release (#335) 2023-05-06 10:32:23 -04:00
Nathaniel Simard a50262b730
refactor: grad clipping (#326) 2023-05-05 12:56:35 -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
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 0d03a2752e
Feat/improve serialization settings (#311) 2023-04-25 12:46:50 -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
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
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 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 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
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
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 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 00625d1527 fix: add version to path dependencies 2023-03-21 10:13:44 -04:00
Nathaniel Simard 4e28e2a776
chore: prepare release v0.6.0 (#246) 2023-03-21 09:47:37 -04:00
Nathaniel Simard c9e344a97f
Fix/batch norm (#238) 2023-03-17 09:31: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 2c58e0d56f
fix(burn-core): load option tensor param (#218) 2023-03-09 15:35:21 -05:00
Nathaniel Simard b34987f1f5
Perf/ndarray ipow (#217) 2023-03-09 15:23:14 -05:00
Dilshod Tadjibaev c5c1104d75
Make parameter initialization configurable (#215) (#216) 2023-03-09 15:21:36 -05:00
Nathaniel Simard d8e5b3fed1
fix(burn-train): use single device loop (#212) 2023-03-08 19:35:10 -05: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