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

57 Commits

Author SHA1 Message Date
Dilshod Tadjibaev e882d41f8b
Use updated serde_rusqlite version (MIT/Apache2 license) (#956) 
fixes #719
 2023-11-14 19:03:07 -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
Louis Fortier-Dubois e4d9d67526
make candle available (#886) 2023-10-23 10:00:39 -04:00
Dilshod Tadjibaev 097fd956d0
Upgrade dependency versions (#854) 
This updates dependencies including tch to 0.14.0, which uses Torch 2.1.
 2023-10-09 14:29:44 -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 80731913c7
Speedup CI (#835) 2023-09-29 15:13:16 -04:00
Nathaniel Simard ca787d6446
Feat/async read (#833) 2023-09-28 17:09:58 -04:00
Louis Fortier-Dubois aa90fe8efb
Refactor/burn benchmark (#829) 2023-09-28 09:38:21 -04:00
Nathaniel Simard ac4adb54ea
Burn compute (#809) 2023-09-18 19:56:53 -04:00
Justin Moore 6095dd104e
Add `cargo-xtask` helper and move scripts into it (#757) 2023-09-06 08:22:00 -04:00
Damien Elmes 3669d2a6d4
Migrate from log4rs to tracing (#739) 2023-08-31 21:07:26 -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
dependabot[bot] 8f4faabd5f
Update gix-tempfile requirement from 7.0.0 to 8.0.0 (#679) 
Updates the requirements on [gix-tempfile](https://github.com/Byron/gitoxide) to permit the latest version.
- [Release notes](https://github.com/Byron/gitoxide/releases)
- [Changelog](https://github.com/Byron/gitoxide/blob/main/CHANGELOG.md)
- [Commits](https://github.com/Byron/gitoxide/commits)

---
updated-dependencies:
- dependency-name: gix-tempfile
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-23 10:32:30 -04:00
Louis Fortier-Dubois c1eddf04fc
Feat/candle/initialize (#650) 2023-08-17 08:50:08 -04:00
Dilshod Tadjibaev 294a319557
Add Jupyter notebook examples (#651) 2023-08-16 18:17:12 -04:00
Dilshod Tadjibaev 5cc32cc8cb
Remove topological sort of nodes (#589) 
ONNX nodes should come as topologically sorted, so we are removing it instead, we are making sure the nodes are topologically sorted.
 2023-08-06 10:51:28 -04:00
Dilshod Tadjibaev 1554a3c898
Full support for ONNX scalar operators and Constants (#578) 2023-08-04 16:51:51 -04:00
dependabot[bot] 739e8d910d
Update sanitize-filename requirement from 0.4.0 to 0.5.0 (#569) 
Updates the requirements on [sanitize-filename](https://github.com/kardeiz/sanitize-filename) to permit the latest version.
- [Commits](https://github.com/kardeiz/sanitize-filename/commits)

---
updated-dependencies:
- dependency-name: sanitize-filename
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-01 08:30:49 -04:00
dependabot[bot] 9d3c9d8fdf
Update wgpu requirement from 0.16.1 to 0.17.0 (#521) 
Updates the requirements on [wgpu](https://github.com/gfx-rs/wgpu) to permit the latest version.
- [Release notes](https://github.com/gfx-rs/wgpu/releases)
- [Changelog](https://github.com/gfx-rs/wgpu/blob/trunk/CHANGELOG.md)
- [Commits](https://github.com/gfx-rs/wgpu/compare/v0.16.1...v0.16.3)

---
updated-dependencies:
- dependency-name: wgpu
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-07-24 13:00:22 -04:00
dependabot[bot] 7d7bd1f135
Update rstest requirement from 0.17.0 to 0.18.0 (#467) 2023-07-05 17:45:47 -04:00
Nathaniel Simard f42176e93a
chore: upgrade syn (#445) 2023-06-28 12:50:28 -04:00
dependabot[bot] d2f2c87120
Update gix-tempfile requirement from 6.0.0 to 7.0.0 (#432) 
Updates the requirements on [gix-tempfile](https://github.com/Byron/gitoxide) to permit the latest version.
- [Release notes](https://github.com/Byron/gitoxide/releases)
- [Changelog](https://github.com/Byron/gitoxide/blob/main/CHANGELOG.md)
- [Commits](https://github.com/Byron/gitoxide/compare/gix-tempfile-v6.0.0...gix-tempfile-v6.0.0)

---
updated-dependencies:
- dependency-name: gix-tempfile
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-06-26 16:39:33 -04:00
Dilshod Tadjibaev 61291cc30c
Revert temp half-rs fix for no_std (#430) 
Fixes #268
 2023-06-26 15:59:06 -04:00
Nathaniel Simard b6f88f8fca
refactor: better templating for wgsl shaders (#404) 2023-06-15 11:59:37 -04:00
Dilshod Tadjibaev d57ca96695
Upgrade dep versions (#399) 2023-06-14 09:55:19 -04:00
Nathaniel Simard 974fdfaba1
Feat/wgpu backend setup (#376) 2023-06-02 11:52:47 -04:00
Dilshod Tadjibaev 483f9acca5
Enhance SqliteDatasetWriter to remove tmp on SIGTERM (#375) 
Temporary files were dangling when a process is terminated by SIGINT (ctrl/cmd+c) or SIGTERM. This change used gix-tempfile to wrap a temporary file with a special signal handle to will remove the tmp file when writer is dropped or when program is terminated early.
 2023-06-02 09:49:48 -04:00
Dilshod Tadjibaev f05cf7bac3
Add SqliteDatasetWriter to store items in sqlite db (#364) 2023-06-01 09:54:50 -04:00
Dilshod Tadjibaev 41e54f741b
Use sqlite bundled version (#366) 
Fixes #365
 2023-05-26 18:09:01 -04:00
Dilshod Tadjibaev b170b539b7
Upgrade dep versions (#359) 2023-05-21 09:07:39 -04:00
Dilshod Tadjibaev 6fece7e4cb
Dataset Improvements: Add Sqlite storage backend and HF importer improvements (#353) 2023-05-20 14:24:55 -04:00
Dilshod Tadjibaev 314db93b7f
Add ability to load onnx state to the generated source code (#319) 2023-05-03 13:05:43 -04:00
Dilshod Tadjibaev df980d534e
Add foundation for importing ONNX files (#297) 2023-04-15 10:44:50 -04:00
Nathaniel Simard 6f43d983f7
State serialization/deserialization overhaul (#247) 2023-03-23 11:02:46 -04:00
Nathaniel Simard eb7b1ceb2f
Bump tch version (#245) 2023-03-20 12:01:55 -04:00
Dilshod Tadjibaev 6222b887e9
Bump to the latest minor versions of dependencies (#237) 2023-03-15 21:54:47 -04: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
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
Dilshod Tadjibaev 9091363ada
Make burn-ndarray and burn-tensor no_std compatible (#168) (#169) 2023-02-21 08:35:24 -05:00
Nathaniel Simard 54c30abfb8
Refactor/burn core (#143) 2023-01-01 19:21:08 -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 713f078602
refactor: burn tensor testgen (#104) 2022-11-16 21:02:32 -05:00
Nathaniel Simard ab51c22a55
Refactor/extract tch backend (#103) 2022-11-15 21:06:40 -05:00
Nathaniel Simard 5b4855317b
Doc/improve example (#64) 2022-10-26 18:08:38 -04:00
Nathaniel Simard bd36023b06
Update projects (#29) 2022-09-04 14:22:56 -04:00
Nathaniel Simard 1ff11e537a
Feat/save load (#27) 2022-09-01 20:31:45 -04:00
Nathaniel Simard 508cfd2b5a
Refactor/metric (#23) 2022-08-23 16:39:30 -04:00
Nathaniel Simard 0aa486fc0c
Feat/logger (#21) 2022-08-23 11:47:21 -04:00
Nathaniel Simard 0f6b50fc10
Feat/dataloader (#17) 2022-08-22 11:21:22 -04:00