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

167 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 3066196297
Fix where sqrt should’ve been powf(2.) (#1052) 2023-12-06 19:02:15 -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
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Nathaniel Simard f6d14f1b1a
Refactor feature flags (#1025) 2023-12-01 09:48:28 -05:00
David Chavez f73136e3df
chore(candle): Allow enabling accelerate (#1009) 
* chore(candle): Allow enabling accelerate

* Temporarily disable test for accelerate feature

* Allow enabling accelerate from upstream

* Update the README

* Have xtask also test using accelerate

* Renable failing test

* Fix matmul on candle when using accelerate

* Add additional comment to xtask method
 2023-11-30 13:03:00 -05:00
Will Brickner 03af140e12
Implement Quiet Softmax (`Attention Is Off By One`) (#692) 
* Added quiet_softmax

* Undid bad formatting

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2023-11-30 12:58:30 -05:00
Nathaniel Simard 630044e96b
Fix/devices api (#990) 2023-11-22 10:24:24 -05:00
Nathaniel Simard 3d6c738776
Refactor/fusion/graph (#988) 2023-11-22 09:55:42 -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 445f41bb7b
Support count_include_pad attr in avg_pool2d ONNX (#978) 
Fixes #636
 2023-11-21 12:21:12 -06:00
David Chavez 88b442015d
feature(norm): Add GroupNorm (#963) 
* Add GroupNorm

* Fix implemenation and add tests

* Address PR comments

* Fix formatting

* Update burn book
 2023-11-21 09:22:56 -05:00
Luni-4 445603401d
ci/Check dependencies (#895) 2023-11-19 10:35:03 -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
Nathaniel Simard 96524d40a1
[Breaking] Refactor Backend Names (#904) 2023-10-29 18:27:49 -04:00
Nathaniel Simard 233922d60c
Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -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
Nathaniel Simard 86db5dc392
Enable candle cuda (#887) 2023-10-23 11:00:54 -04:00
Louis Fortier-Dubois e4d9d67526
make candle available (#886) 2023-10-23 10:00:39 -04:00
Nathaniel Simard d263968236
Refactor unfold4d + Add Module (#870) 2023-10-22 11:53:59 -04:00
Mathias Insley 1962c06c21
Bug/lstm unsqueeze (#873) 2023-10-18 18:53:37 -04:00
Nathaniel Simard 6c125e161f
Fix CI Rust Update 1.73 (#849) 2023-10-06 12:53:45 -04:00
Nathaniel Simard 904ff1a974
Refactor burn-train (#847) 2023-10-05 13:10:54 -04:00
Eddie Mata Ewy 9afc76303f
removed batch_size field from GRU and LSTM (#845) 
* removed batch_size field from GRU and LSTM

* slight style difference

---------

Co-authored-by: Eduardo Mata Ewy <eddiemata@hotmail.es>
 2023-10-05 11:15:20 -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 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 bdcd5873c5
Support no-std for models generated by burn-import (#822) 2023-09-21 11:43:07 -04:00
Dilshod Tadjibaev 35ca839e18
Add support for different record types in ONNX (#816) 2023-09-21 09:06:57 -04:00
Nathaniel Simard 57d6a566be
Feat/dashboard tui (#790) 2023-09-13 10:45: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 76894eff04 Remove candle from release 2023-09-06 10:25:37 -04:00
Arvid Hammarlund f330f9f73d
Added Dim-iter (#771) 
* "added dim-iter".

* added tests to iter_dim

* Made the rnn implemenations use squeeze.
 2023-09-06 09:22:30 -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 a20a1a8cee
Fixed bug in binary cross entropy (#772) 2023-09-05 17:07:07 -04:00
Nathaniel Simard a9596af07a
Doc: improve config builder docs (#760) 
* Doc: improve config builder docs

* fix ci

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-09-05 10:16:55 -04:00
Nathaniel Simard c484999d54
Fix CI (#766) 2023-09-04 20:55:31 -04:00
Arvid Hammarlund b58af4a4a3
Cross Entropy with label smoothing and weights. (#755) 2023-09-04 20:20:47 -04:00
Nathaniel Simard 36446e8d35
fix: rnn mut (#764) 2023-09-04 10:09:15 -04:00
Damien Elmes 5b97f1a54b
Re-export some submodules and add feature flags for them (#759) 2023-09-04 09:13:52 -04:00
Damien Elmes 520cea16df
Fix: burn-core/std depended on sqlite feature (#731) 2023-09-01 09:28:58 -04:00
Damien Elmes 0fd82ee5d3
Allow opting out of SQLite's bundling, or all SQLite usage (#711) 2023-08-29 10:50:01 -04:00
Nathaniel Simard f024dc9ccb
fix: dataloader reshuffle dataset each iter (#703) 2023-08-28 12:48:33 -04:00
Damien Elmes 9f1a68bc33
Fix: processed count was one greater than total items at completion (#708) 2023-08-28 11:56:42 -04:00
Asuka Minato 0f7864f11b
Path type (#699) 2023-08-27 09:17:48 -04:00
Will Brickner bab036b263
Added custom initializer option for MHA and other transformer-related modules (#693) 2023-08-25 09:55:05 -04:00