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

126 Commits

Author SHA1 Message Date
yurzhang 419e53bc42
Support signed value for `Tensor::arange` (#1238) 2024-02-07 09:33:01 -05:00
Sylvain Benner c4e512d176
Create doc feature in all burn crates and use them for docs.rs build (#1212) 2024-02-01 12:47:46 -05:00
Nathaniel Simard b8915e91b7
Fix problems with batch norm on LibTorch backend (#1226) 
* Fix problems with batch norm

* Add comment
 2024-02-01 12:26:20 -05:00
Sylvain Benner 4aa13d6b25
Bump Burn version to 0.13 (#1211) 2024-01-31 16:01:20 -05:00
Nathaniel Simard 2acf6561dc
Chore: put all dependencies versions in workspace (#1210) 2024-01-31 14:47:02 -05:00
Louis Fortier-Dubois e03facc5de
fix powf bugs (#1207) 2024-01-31 11:38:45 -05:00
Guillaume Lagrange f1d98bc5f8
Bump tch version to 0.15.0 with libtorch 2.2.0 (#1206) 2024-01-31 11:29:44 -05:00
Guillaume Lagrange de84b5c457
Add setup instructions for burn-tch (#1185) 2024-01-31 10:18:09 -05:00
Joshua Ferguson 4a70a0f8bc
renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 2024-01-27 10:04:50 -05:00
Joshua Ferguson 3b7d9feede
Elementwise pow op (#1133) 2024-01-24 09:46:57 -05:00
Sylvain Benner 707b61051c
Fix doc build on docs.rs (#1168) 2024-01-23 14:13:57 -05:00
wcshds a5bdf38c92
fix the problem of sigmoid gradient generating NaN (#1140) 
* use sigmoid derivative formulas

* add test

* fix test error

* move sigmoid to tensor/ops/activation.rs

* use full precision in the default implementation

* rename the param of `sigmoid_backward`
 2024-01-16 16:20:18 -05:00
Kirill Mavreshko 97297538b1
Remove _devauto fuctions (#518) (#1110) 2024-01-06 13:36:34 -05:00
Nathaniel Simard d82e6b157b
Fix tests (#1089) 
* Fix tests

* Fix fmt

* Fix CI
 2023-12-21 13:06:19 -05:00
Kirill Mavreshko 1fd07fcb4a
Explicit device tensors (#1081) 2023-12-20 17:49:59 -05:00
Kelvin Wu 7c6f017c98
Implement chunk for different backends (#1032) 2023-12-20 13:35:59 -05:00
Nathaniel Simard 2ca1fd5a92
Fix bugs (#1064) 2023-12-12 16:19:01 -05:00
David Chavez 71d3c1d142
chore(infra): Share some properties across workspace (#1039) 2023-12-12 09:39:07 -05:00
Louis Fortier-Dubois 8fc52113bc
Chore/bump v12 (#1048) 2023-12-04 10:47:54 -05:00
Louis Fortier-Dubois 3088c466a5
patch 0.11.1 (#1047) 2023-12-04 10:18:30 -05:00
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Nathaniel Simard b0de56da29
Fix double broadcast with tch (#1026) 
* Fix double broadcast with tch

* More fixes

* Fix clippy warm
 2023-12-01 10:02:57 -05:00
Nathaniel Simard f6d14f1b1a
Refactor feature flags (#1025) 2023-12-01 09:48:28 -05:00
Nathaniel Simard 3d6c738776
Refactor/fusion/graph (#988) 2023-11-22 09:55:42 -05:00
Dilshod Tadjibaev cb616ed72c
Fix image classifier build failure by upgrading versions (#967) 2023-11-21 09:28:56 -05:00
Nathaniel Simard 8f1526b9c9
Update readme (#962) 2023-11-17 13:04:41 -05:00
Zsombor 4fc0c27e31
Implement tensor.recip() function to calculate elementwise reciprocals (#953) 2023-11-15 09:17:32 -05:00
Nathaniel Simard 96524d40a1
[Breaking] Refactor Backend Names (#904) 2023-10-29 18:27:49 -04:00
Nathaniel Simard 233922d60c
Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -04:00
Nathaniel Simard d263968236
Refactor unfold4d + Add Module (#870) 2023-10-22 11:53:59 -04:00
Mathias Insley 255dfefab2
Feat/tensor unfold (#819) 2023-10-15 17:05:34 -04:00
Dilshod Tadjibaev 097fd956d0
Upgrade dependency versions (#854) 
This updates dependencies including tch to 0.14.0, which uses Torch 2.1.
 2023-10-09 14:29:44 -04:00
Nathaniel Simard ca787d6446
Feat/async read (#833) 2023-09-28 17:09:58 -04:00
Louis Fortier-Dubois aa90fe8efb
Refactor/burn benchmark (#829) 2023-09-28 09:38:21 -04:00
Nathaniel Simard 95e660488e
Refactor/burn compute wgpu (#826) 2023-09-25 10:42:45 -04:00
Louis Fortier-Dubois 8c215e8be3
Bugfix/int swap dims (#823) 2023-09-22 08:38:38 -04:00
Juliano Decico Negri 293020aae6
#384 Include tests for int.rs and float.rs (#794) 2023-09-21 09:00:09 -04:00
Nathaniel Simard af0be5cfeb
Chore: bump version (#777) 2023-09-06 12:15:13 -04:00
Louis Fortier-Dubois 760c9e1d8e
Feat/candle/module ops (#725) 2023-08-30 18:53:03 -04:00
Louis Fortier-Dubois 7c34e21424
Perf/tensor ops/more tests (#718) 2023-08-30 09:08:18 -04:00
Louis Fortier-Dubois c89f9969ed
Perf/tensor ops/tests (#710) 2023-08-28 12:53:17 -04:00
MOZGIII 7f558bdc46
Expose element traits (#700) 2023-08-27 09:02:39 -04:00
Nathaniel Simard efee0ac296
Feat/train/custom optimize method (#689) 
* Add the possibility to add a custom optimize function for models

* Fix clippy
 2023-08-25 07:14:36 -04:00
Louis Fortier-Dubois bc27a87e9d
new clippy stuff (#687) 2023-08-24 13:20:58 -04:00
Will Brickner d1a708e317
Made TchTensor fields public (#677) 2023-08-22 21:37:58 -04:00
Caio Piccirillo 2fefc82099
Dilation maxpool (#668) 2023-08-21 14:14:25 -04:00
Nathaniel Simard bda03c6a76
Feat/avg pool/include pad config (#653) 2023-08-17 08:50:31 -04:00
Louis Fortier-Dubois c1eddf04fc
Feat/candle/initialize (#650) 2023-08-17 08:50:08 -04:00
Caio Piccirillo cb283a9e5b
Max pool1d (#602) 2023-08-09 16:13:48 -04:00
Caio Piccirillo 1d3bbaab13
Typos (#608) 2023-08-08 17:57:51 -04:00