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

264 Commits

Author SHA1 Message Date
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
Dilshod Tadjibaev 4e9e6d2706
Move unsqueeze op to the tensor's base (#261) 2023-04-01 14:20:48 -04:00
Nathaniel Simard 32d38bebc3
Refactor Param wrapping only for Tensor (#259) 2023-03-31 16:45:10 -04:00
Dilshod Tadjibaev 7364d09d32
Add flatten op to the tensor base (#260) 2023-03-31 16:44:48 -04:00
Nathaniel Simard c14c7977ec
Text classification inference example (#256) 2023-03-26 16:51:37 -04:00
Nathaniel Simard ed24db6d3e
Fix dataset clash (#255) 2023-03-25 11:50:06 -04:00
Nathaniel Simard 8d03fc2e90
Inline serde struct (#252) 2023-03-24 17:58:48 -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 a74e4cd0bc
fix: conv bias backward (#248) 2023-03-23 16:10:26 -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 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