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

164 Commits

Author SHA1 Message Date
Alex Errant 610d64095e
cargo +nightly fmt (#1017) 2023-12-12 13:29:06 -05:00
David Chavez 71d3c1d142
chore(infra): Share some properties across workspace (#1039) 2023-12-12 09:39:07 -05:00
David Chavez ef09b637b9
chore(web): better panic (#1051) 2023-12-05 07:30:29 -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
Timothy 52811f9938
Fix some issues in Burn book (#1042) 2023-12-04 09:23:47 -05:00
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Dilshod Tadjibaev 4192490b88
Update with demo links (#1033) 2023-12-01 13:07:13 -05:00
Nathaniel Simard 12911c3ad6
Fix/mnist inference web (#1029) 2023-12-01 12:05:31 -05:00
Nathaniel Simard f6d14f1b1a
Refactor feature flags (#1025) 2023-12-01 09:48:28 -05:00
Luni-4 ba1de9c654
Ci/Schedule dependencies checks (#1014) 
* ci: Add audit file

* ci: Schedule dependencies checks

* mnist-inference-web: Remove unused dep

* Run when a new release is created

* Fix comment
 2023-11-30 13:06:27 -05:00
Nathaniel Simard 630044e96b
Fix/devices api (#990) 2023-11-22 10:24:24 -05:00
Nathaniel Simard cabbaab0c4
Fix/constant tensors (#984) 
* Generalize autodiff tensor

* Can have int const module

* Update example

* Support no-std with burn-import

* Fix typos

* Fix alloc problems

* Revert burn-import changes

* Fix examples

* Support Int and Bool Params

* Fix

* Add comment
 2023-11-21 15:27:28 -06:00
Dilshod Tadjibaev cb616ed72c
Fix image classifier build failure by upgrading versions (#967) 2023-11-21 09:28:56 -05:00
Luni-4 445603401d
ci/Check dependencies (#895) 2023-11-19 10:35:03 -05:00
Luni-4 ec9df53d4c
ci/Fix `cargo clippy` action (#942) 2023-11-16 19:35:38 -05:00
Nathaniel Simard 322480b744
Feat/op fusion decorator (#939) 
* WIP

* Impl backend decorator

* WIP

* WIP

* WIP

* WIP

* WIP

* WIP

* Refactor

* Handle graph single ops execution

* WIP

* Starting to get concrete

* WIP

* Fix locator

* Implement add ops

* Start implementing ops

* Add more ops

* Add more ops

* More float ops

* Almost finish float ops

* Almost done with Int

* Some fix

* Into float

* Implement bool ops

* Almost done with MVP

* Fix adaptive pooling

* Add fusion as backend

* Fix memory leak

* Fix

* WIP Doc

* Doc all ops enum

* Initial docs

* Clippy

* Clippy v2

* Fix typos

* Fix doc

* Fix feature flags

* Add missing ops

* Some cleanup

* Revert u128 id

* cosmetic fixes

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-11-09 21:21:41 -05:00
Aisuko 4e8b573f13
Fixed the wrong order of the attributes (#930) 
Signed-off-by: GitHub <noreply@github.com>
 2023-11-03 09:21:58 -04:00
Louis Fortier-Dubois 8742d31d16
Perf/wgpu/matmul vec4rhs (#914) 2023-10-31 08:37:17 -04:00
Nathaniel Simard 96524d40a1
[Breaking] Refactor Backend Names (#904) 2023-10-29 18:27:49 -04:00
Louis Fortier-Dubois e2a3329997
Feat/wgpu/autotune compute (#906) 2023-10-29 16:44:59 -04:00
Luni-4 7332ebcabf
ci/Add coverage as xtask task (#902) 2023-10-26 12:45:08 -04:00
Nathaniel Simard 233922d60c
Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -04:00
Louis Fortier-Dubois e76b6d47de
WGPU: matmul vec4 (#897) 2023-10-24 17:23:43 -04:00
Louis Fortier-Dubois 0ab611b42e
AdamW NaN fix (#888) 2023-10-24 14:48:40 -04:00
Alex Errant 9f2bc599b8
Add a `sync` feature to common, core, and tensor (#893) 2023-10-24 14:32:01 -04:00
Louis Fortier-Dubois d96f73da0a
Feat/compute/autotune (#861) 
* wip autotune compute

* too much generics

* wip

* megawip

* in progress

* first test passes

* first test passes

* fixed test

* refactor for cache hit and miss

* cleanup and fixes

* doc and stuff

* doc and stuff

* clippy

* format

* remove lifetime

* cleanup operation

* wip

* wip

* compiles

* wip mutable borrow

* refactor with autotune server

* wip tune benchmark

* test passes

* fix autotune key

* cache hit miss tests

* refactor wgpu to match burn-compute

* better operation execution

* cleanup & refactor

* test for parametered kernel

* fmt

* fmt

* clippy

* allow clippy

* fix no-std

* fmt

* review and ci

* Fix CI

* delete dummy benchmarks again

---------

Co-authored-by: nathaniel <nathaniel.simard.42@gmail.com>
 2023-10-23 11:29:44 -04:00
Louis Fortier-Dubois e4d9d67526
make candle available (#886) 2023-10-23 10:00:39 -04:00
Nathaniel Simard af813d09ed
Feat/early stopping + burn train refactor (#878) 2023-10-20 11:47:31 -04:00
Nathaniel Simard dd4e72a98f
Feat/checkpoint criteria (#862) 
* WIP

* Setup

* Test metrics

* Fix bug

* Cleanup
 2023-10-17 09:03:11 -04:00
Nathaniel Simard 620b86de98
Feat training events (#857) 2023-10-10 13:27:03 -04:00
Nathaniel Simard 904ff1a974
Refactor burn-train (#847) 2023-10-05 13:10:54 -04:00
Dilshod Tadjibaev e2a17e4295
Add image classification web demo with WebGPU, CPU backends (#840) 2023-10-05 10:29:13 -04:00
Nathaniel Simard ce120ead3a
Improve metrics (#844) 2023-10-03 18:15:43 -04:00
Daniel Vainsencher 1e609acfe5
Bugfix: epoch and iteration were swapped. (#838) 
Co-authored-by: Daniel Vainsencher <danielv@nirinor.ai>
 2023-10-01 09:48:51 -04:00
Nathaniel Simard ca787d6446
Feat/async read (#833) 2023-09-28 17:09:58 -04:00
Nathaniel Simard 95e660488e
Refactor/burn compute wgpu (#826) 2023-09-25 10:42:45 -04:00
Dilshod Tadjibaev 35ca839e18
Add support for different record types in ONNX (#816) 2023-09-21 09:06:57 -04:00
Damien Elmes d7e9e75099
Fix train-minimal feature and ensure it gets tested (#802) 2023-09-16 09:52:14 -04:00
Nathaniel Simard 4f72578260
Fix compilation flags (#800) 2023-09-13 10:38:16 -04:00
Nathaniel Simard af0be5cfeb
Chore: bump version (#777) 2023-09-06 12:15:13 -04:00
Nathaniel Simard 8b3d10c4d3
Cleanup of the book + feature flags (#773) 
---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-09-06 09:16:36 -04:00
Arvid Hammarlund b58af4a4a3
Cross Entropy with label smoothing and weights. (#755) 2023-09-04 20:20:47 -04:00
Damien Elmes 5b97f1a54b
Re-export some submodules and add feature flags for them (#759) 2023-09-04 09:13:52 -04:00
Louis Fortier-Dubois 56b8e29514
Doc/building blocks more (#745) 2023-09-02 13:09:03 -04:00
Damien Elmes 520cea16df
Fix: burn-core/std depended on sqlite feature (#731) 2023-09-01 09:28:58 -04:00
Nathaniel Simard c95b34c511
Book: backend extension + custom wgpu kernel (#728) 2023-08-31 09:55:43 -04:00
Asuka Minato 0f7864f11b
Path type (#699) 2023-08-27 09:17:48 -04:00
Louis Fortier-Dubois bc27a87e9d
new clippy stuff (#687) 2023-08-24 13:20:58 -04:00
Nathaniel Simard 00d3d208b8
Add book + fix some code (#671) 2023-08-23 11:52:55 -04:00