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

60 Commits

Author SHA1 Message Date
Nathaniel Simard 2a4ba5a6ab
Feat/gather scatter (#367) 2023-05-27 11:40:04 -04:00
Dilshod Tadjibaev b170b539b7
Upgrade dep versions (#359) 2023-05-21 09:07:39 -04:00
Nathaniel Simard 976102fec0
Feat/avg pool1d (#349) 2023-05-15 08:29:45 -04:00
Dilshod Tadjibaev fbbc8ac560
Remove build panic from burn-tch (#347) 2023-05-12 09:37:58 -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 Simard 29eecd6383
Prepare next release (#335) 2023-05-06 10:32:23 -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
Nathaniel Simard b54f9302c7
Feat/avg pool2d (#318) 2023-04-30 12:25:14 -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
Yu Sun 26cf555612
Feat: add new tensor ops mask_scatter (#258) 2023-04-13 08:58:31 -04:00
Nathaniel Simard 04bcf9550a
Fix/text gen example (#292) 2023-04-11 17:18:45 -04:00
Nathaniel Simard f04fe101d8
Feat/module no grad (#274) 2023-04-07 09:01:27 -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 eb7b1ceb2f
Bump tch version (#245) 2023-03-20 12:01:55 -04:00
Nathaniel Simard c9e344a97f
Fix/batch norm (#238) 2023-03-17 09:31:52 -04:00
Nathaniel Simard d09ab44979
Feat/index_select (#227) 2023-03-12 17:44:22 -04:00
Nathaniel Simard 9655b74b22
Feat/index_select_dim ops (#225) 2023-03-11 16:14:57 -05:00
Nathaniel Simard 0544a915eb
Refactor/migrate more numeric func (#220) 2023-03-10 10:47:15 -05:00
Nathaniel Simard cf7847acb5
fix: mix precision training on tch backend (#209) 2023-03-07 09:42:51 -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
Nathaniel Simard 02591e5a6f
fix(burn-tch): inplace binary ops (#193) 2023-03-05 11:23:57 -05:00
Nathaniel Simard 15ec42dd6f
Refactor/backend bool tensor (#192) 2023-03-05 11:23:46 -05: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 a62738b0f4
Add a workaround script for arm64 tch-rs build issue (#180) (#182) 
* Add a workaround script for arm64 tch-rs build issue (#180)
 2023-02-28 20:01:01 -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 23a58acfdb
fix: arange poor performance (#167) 2023-02-18 15:56:32 -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 2401d8ad96
Prepare next release (#161) 2023-02-12 15:32:29 -05:00
Yu Sun 0b85cb0eed
feat(trait-TensorOps): add log1p (#160) 2023-02-11 13:30:50 -05:00
Nathaniel Simard c7963d8485
refactor: device functions (#157) 2023-01-27 18:37:21 -05:00
Nathaniel Simard 2d4e514b41
Refactor/shape function (#156) 2023-01-27 15:18:55 -05:00
Makro f6f0d0e4f3
Add cos, sin and tanh operations (#155) 
* Add cos, sin and tanh operations

* Add tests

* Fix formatting
 2023-01-24 19:40:30 -05:00
Nathaniel Simard 34d233cd3e
Feat/max pooling backend (#152) 2023-01-21 15:39:21 -05:00
Nathaniel Simard 745c88f0a0
Feat/conv (#147) 2023-01-11 18:33:09 -05:00
Nathaniel Simard 2f179f12c9
Bump versions (#141) 2022-12-30 15:15:51 -05:00
Nathaniel Simard eea5a263bf
Feat/adam optimizer (#140) 2022-12-30 15:02:43 -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 d9592411c2
Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00