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

1395 Commits

Author SHA1 Message Date
Nathaniel Simard 7d2f43dfca
Refactor Tensor API (#163) 2023-02-17 17:31:20 -05:00
nathaniel 54e2c1a600 chore: add doc issue template 2023-02-16 09:09:53 -05:00
Aaron Roney f907d0025d
Add feature to load config / state from binary slice. (#164) 2023-02-16 08:43:43 -05:00
Nathaniel Simard 2401d8ad96
Prepare next release (#161) 2023-02-12 15:32:29 -05:00
nathaniel 2c4fc72c47 Update README 2023-02-12 15:17:30 -05:00
Yu Sun 0b85cb0eed
feat(trait-TensorOps): add log1p (#160) 2023-02-11 13:30:50 -05:00
Nathaniel Simard 993a6e3095
Feat/batch norm (#158) 2023-02-02 17:22:40 -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 b7aa066e5c
Feat/nn max pool2d (#153) 2023-01-21 17:06:39 -05:00
Nathaniel Simard 34d233cd3e
Feat/max pooling backend (#152) 2023-01-21 15:39:21 -05:00
Nathaniel Simard 1d0d92a269
Feat/nn conv (#148) 2023-01-12 20:22:48 -05:00
Nathaniel Simard 745c88f0a0
Feat/conv (#147) 2023-01-11 18:33:09 -05:00
nathaniel cadf494f1e Update README.md 2023-01-02 10:40:30 -05:00
Nathaniel Simard 54c30abfb8
Refactor/burn core (#143) 2023-01-01 19:21:08 -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
Nathaniel Simard 248039da0a
Refactor/metric adaptor (#139) 2022-12-26 16:30:25 -05:00
Visual 567adfb93e
refactor: fix all clippy warnings (#137) 2022-12-25 11:22:25 -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 1ec35a9e1b
feat: from floats (#87) 2022-12-25 10:11:20 -05:00
Nathaniel Simard 3a9dfe6097
feat: cross entropy loss (#130) 2022-12-25 10:10:22 -05:00
Nathaniel Simard 1a1d86dc3e
refactor: save and load state (#129) 2022-12-24 13:02:37 -05:00
Nathaniel Simard 3a91c2c48e
Feat/multi device (#128) 2022-12-20 18:01:58 -05:00
Nathaniel Simard a599eaed88
Feat/module visitor (#127) 2022-12-17 14:25:36 -05:00
Nathaniel Simard d9592411c2
Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00
Nathaniel Simard 63d8d39517
Feat/autoregressive transformer (#125) 2022-12-10 15:47:57 -05:00
Nathaniel Simard b99b23e1a7
refactor: backends (#124) 2022-12-02 19:28:34 -05:00
Nathaniel Simard eee90a5c9e
Example/text classification (#123) 2022-12-02 17:42:49 -05:00
Nathaniel Simard 7c38a980c1
feat: improve bool tensor (#122) 2022-12-01 19:43:36 -05:00
Jakub Lewandowski cdc29a02d2
Change links in README to work on crates.io (#114) 2022-11-28 20:08:36 -05:00
Nathaniel Simard 9cade192fd
Feat/transformer encoder (#119) 2022-11-26 20:06:43 -05:00
Nathaniel Simard 8bd0b17296
Feat/mha (#118) 2022-11-26 15:48:26 -05:00
Nathaniel Simard 46d06f0c90
feat: module init (#117) 2022-11-25 22:02:26 -05:00
Nathaniel Simard 4abc281dae
Refactor/forward pass (#116) 2022-11-25 21:14:44 -05:00
Nathaniel Simard acb14adc29
Fix/named tensor (#115) 2022-11-25 19:30:35 -05:00
Nathaniel Simard e0e787f87d
Experimental/named tensor (#113) 2022-11-23 19:05:46 -05:00
Nathaniel Simard c4c739d91b
doc: update readme (#112) 2022-11-20 19:41:55 -05:00
nathaniel df95fc38ae ci: cleanup dependency graph 2022-11-20 13:26:30 -05:00
nathaniel a8e75f6164 fix/publish-crates-io 2022-11-20 13:08:54 -05:00
Nathaniel Simard 9ecd1be992
chore: get ready for next release (#111) 2022-11-20 12:59:10 -05:00
Nathaniel Simard 03b216a318
Ci/use template (#110) 2022-11-20 12:29:02 -05:00
Nathaniel Simard cd39a1c4cd
Refactor/autodiff graph (#108) 2022-11-19 20:15:12 -05:00
Nathaniel Simard ca94a9f105
refactor: autodiff gradients types (#107) 2022-11-19 19:43:49 -05:00
Nathaniel Simard dda067e79b
Refactor/backend autodiff (#106) 2022-11-19 12:37:06 -05:00
Nathaniel Simard d45d674a04
Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
Nathaniel Simard 713f078602
refactor: burn tensor testgen (#104) 2022-11-16 21:02:32 -05:00
Nathaniel Simard ab51c22a55
Refactor/extract tch backend (#103) 2022-11-15 21:06:40 -05:00
Nathaniel Simard 23677b8e89
Refactor/zeros ones elems (#102) 2022-11-14 18:41:26 -05:00