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
|
403500f018
|
feat(burn-core): refactor cross entropy (#229)
|
2023-03-13 13:47:19 -04: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
|
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
|
c510e8bfb5
|
Add accelerate framework support for burn-ndarray (#183) (#184)
|
2023-02-28 09:09:47 -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
|
7d2f43dfca
|
Refactor Tensor API (#163)
|
2023-02-17 17:31:20 -05:00 |
Nathaniel Simard
|
c7963d8485
|
refactor: device functions (#157)
|
2023-01-27 18:37:21 -05:00 |
Nathaniel Simard
|
b7aa066e5c
|
Feat/nn max pool2d (#153)
|
2023-01-21 17:06:39 -05:00 |
Nathaniel Simard
|
eea5a263bf
|
Feat/adam optimizer (#140)
|
2022-12-30 15:02:43 -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
|
3a9dfe6097
|
feat: cross entropy loss (#130)
|
2022-12-25 10:10:22 -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
|
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
|
c4c739d91b
|
doc: update readme (#112)
|
2022-11-20 19:41:55 -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
|
ab51c22a55
|
Refactor/extract tch backend (#103)
|
2022-11-15 21:06:40 -05:00 |
Nathaniel Simard
|
3da122db09
|
Add cuda gpu example + doc (#91)
|
2022-11-09 21:32:51 -05:00 |
Nathan Henrie
|
947ed00301
|
Fix `cargo run --example mnist` (#90)
|
2022-11-09 20:45:58 -05:00 |
Nathaniel Simard
|
c5213b6c32
|
Fix: mnist example (#85)
|
2022-11-07 17:58:50 -05:00 |
Nathaniel Simard
|
5b4855317b
|
Doc/improve example (#64)
|
2022-10-26 18:08:38 -04:00 |