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

250 Commits

Author SHA1 Message Date
nathaniel ef92d8dd1a fix: missing info in cargo.toml 2023-03-21 09:56:34 -04:00
Nathaniel Simard 4e28e2a776
chore: prepare release v0.6.0 (#246) 2023-03-21 09:47:37 -04:00
nathaniel 2f91a5e4e2 fix: burn mnist wasm demo link 2023-03-20 19:31:56 -04:00
Nathaniel Simard eb7b1ceb2f
Bump tch version (#245) 2023-03-20 12:01:55 -04:00
Dilshod Tadjibaev 019ccbde4c
Fix and update readme docs (#244) 2023-03-20 11:51:07 -04:00
Dilshod Tadjibaev 8e74ce4bb9
Reduce MNIST parameters and update mnist-inference-web to match (#242) 2023-03-17 19:46:26 -04:00
Nathaniel Simard 04d72631d7
Doc/update readme (#241) 2023-03-17 13:46:36 -04:00
Dilshod Tadjibaev 608ee3f124
Fix discord link (fixes #239) (#240) 
The previous link stopped working. This link does not expire or have other limits.
 2023-03-17 09:32:22 -04:00
Nathaniel Simard c9e344a97f
Fix/batch norm (#238) 2023-03-17 09:31:52 -04:00
Dilshod Tadjibaev 6222b887e9
Bump to the latest minor versions of dependencies (#237) 2023-03-15 21:54:47 -04:00
Dilshod Tadjibaev aa8c96d3fb
Update readme with blas-accelerate flag information (#236) 2023-03-15 21:44:13 -04:00
Dilshod Tadjibaev ed5888e701
Mnist inference web: Readme (#233) 2023-03-15 08:49:59 -04:00
Nathaniel Simard a314c7b694
Perf/optimize backward ops (#232) 2023-03-14 12:59:52 -04:00
Dilshod Tadjibaev e991ba38df
Mnist inference web: Remove wee_alloc allocator and use rust's alloc for WASM32 (#231) 2023-03-14 08:57:10 -04:00
nathaniel 7f3f8d2b22 fix: mnist inference web build for web no such dir 2023-03-13 20:13:39 -04:00
Dilshod Tadjibaev bf9d33e6fc
Add MNIST inference on the web demo crate (#228) 
* Add MNIST inference on the web demo crate

* Fix problems identified during a PR review
 2023-03-13 19:51:32 -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 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 860051ca5c
Perf/ndarray maxpool (#223) 2023-03-10 19:14:18 -05:00
Hariganesh Srinivasan 740b554047
Implemented sigmoid and log_sigmoid, Resolves #171 (#221) 
* implemented sigmoid and log_sigmoid with tests

* added test for overflow and computation.
 2023-03-10 19:03:01 -05:00
Nathaniel Simard 06c1997559
refactor(burn-ndarray): use single par iter in conv2d (#222) 2023-03-10 13:16:57 -05:00
Nathaniel Simard 0544a915eb
Refactor/migrate more numeric func (#220) 2023-03-10 10:47:15 -05:00
Nathaniel Simard a6a49cdc2a
perf: improve conv2d performance (#219) 2023-03-10 10:46:22 -05: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 a2ec774c37
draft: Perf/ndarray matmul (#214) 2023-03-09 14:00:35 -05:00
Nathaniel Simard d8e5b3fed1
fix(burn-train): use single device loop (#212) 2023-03-08 19:35:10 -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 b8d9f012da
refactor(burn-core): module visitor mut (#195) 2023-03-05 14:40:47 -05:00
Nathaniel Simard 2318434caf
fix(burn-autodiff): track to device ops (#194) 2023-03-05 11:24:10 -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 bddc1461ef
fix: running state (#190) 2023-03-01 17:33:32 -05:00
Nathaniel Simard d4c298c221
Refactor/burn core (#188) 2023-03-01 16:10:58 -05:00
Dilshod Tadjibaev 30f6a30228
Fix tch build issue from the root (#180) (#189) 
Moved .cargo/config.toml to the root so that the project can be built from the root and burn-tch as well.
 2023-03-01 15:54:37 -05:00
Nathaniel Simard e6e7f4de42
feat: inplace tensor api. (#187) 2023-03-01 10:55:51 -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 37806b576c
Fix FailToDownloadPythonDependencies error (#82) (#185) 2023-02-28 09:23:49 -05:00
Dilshod Tadjibaev c510e8bfb5
Add accelerate framework support for burn-ndarray (#183) (#184) 2023-02-28 09:09:47 -05:00
Dilshod Tadjibaev 9df3b6a3be
Add new packages and tests with no-std options in CI (#168) (#179) 2023-02-26 14:18:13 -05:00
Dilshod Tadjibaev b7be9cec15
Fix CI template that was causing error (#168) (#178) 2023-02-26 13:00:56 -05:00
Dilshod Tadjibaev 83e43a3eac
Add no-std steps to the test template (#168) (#177) 2023-02-26 12:19:47 -05:00
Nathaniel Simard 6ba9de2868
perf: remove unecessary masking in dropout (#172) 2023-02-26 11:43:46 -05:00