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

900 Commits

Author SHA1 Message Date
Sylvain Benner c4e512d176
Create doc feature in all burn crates and use them for docs.rs build (#1212) 2024-02-01 12:47:46 -05:00
Nathaniel Simard b8915e91b7
Fix problems with batch norm on LibTorch backend (#1226) 
* Fix problems with batch norm

* Add comment
 2024-02-01 12:26:20 -05:00
Nathaniel Simard bdc8260905
Fix pow autodiff (#1225) 2024-02-01 12:23:12 -05:00
Nathaniel Simard abb7ee10a5
Fix wgpu performance issue: revert to wgpu 0.18.0 (#1221) 
* Use wgpu 0.18.0

* Fix clippy
 2024-02-01 10:55:08 -05:00
Sylvain Benner 4aa13d6b25
Bump Burn version to 0.13 (#1211) 2024-01-31 16:01:20 -05:00
Nathaniel Simard 2acf6561dc
Chore: put all dependencies versions in workspace (#1210) 2024-01-31 14:47:02 -05:00
Louis Fortier-Dubois e03facc5de
fix powf bugs (#1207) 2024-01-31 11:38:45 -05:00
Guillaume Lagrange f1d98bc5f8
Bump tch version to 0.15.0 with libtorch 2.2.0 (#1206) 2024-01-31 11:29:44 -05:00
Aasheesh Singh c620176943
doc fix for mha weights (#1205) 2024-01-31 11:26:38 -05:00
Nathaniel Simard 0ec0fba869
Chore: Update ratatui version (#1204) 2024-01-31 11:26:09 -05:00
dependabot[bot] 22d97e5a59
Bump web-time from 0.2.4 to 1.0.0 (#1199) 
Bumps [web-time](https://github.com/daxpedda/web-time) from 0.2.4 to 1.0.0.
- [Release notes](https://github.com/daxpedda/web-time/releases)
- [Changelog](https://github.com/daxpedda/web-time/blob/main/CHANGELOG.md)
- [Commits](https://github.com/daxpedda/web-time/compare/v0.2.4...v1.0.0)

---
updated-dependencies:
- dependency-name: web-time
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2024-01-31 11:02:35 -05:00
dependabot[bot] d77b189afb
Bump serial_test from 2.0.0 to 3.0.0 (#1203) 
Bumps [serial_test](https://github.com/palfrey/serial_test) from 2.0.0 to 3.0.0.
- [Release notes](https://github.com/palfrey/serial_test/releases)
- [Commits](https://github.com/palfrey/serial_test/compare/v2.0.0...v3.0.0)

---
updated-dependencies:
- dependency-name: serial_test
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2024-01-31 11:02:17 -05:00
Joshua Ferguson 07c69bd29a
Powf fix2 (#1193) 2024-01-31 10:19:15 -05:00
Guillaume Lagrange de84b5c457
Add setup instructions for burn-tch (#1185) 2024-01-31 10:18:09 -05:00
Dilshod Tadjibaev 14aad24c19
Regex fix pytorch (#1196) 2024-01-31 09:53:52 -05:00
Luni-4 c92c51d8b9
ci: Enable dependabot daily (#1195) 2024-01-31 09:48:01 -05:00
Sylvain Benner 02259ea11c
[xtask] New subcommand: Books (#1192) 2024-01-31 09:09:22 -05:00
amfaber ff222b06b5
[burn-train] Fix keys not working reliably to change metrics in terminal UI (#1101) 
* Exit event handler if KeyEvent is of release kind

* Switch logic to return on press

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2024-01-30 22:26:47 -05:00
Nathaniel Simard b7486b0e80
Feat/fusion/wgpu autotune (#1188) 
* wip

* WIP

* Update

* USe read write lock

* Refactor autotune

* Cleanup

* Add docs
 2024-01-30 11:01:43 -05:00
Aasheesh Singh 8b4038d004
Merge pull request #1183 from tracel-ai/docs/guide 
Docs/guide

Improve guide book: Tensor, Dataset and Example Section.
 2024-01-30 10:25:29 -05:00
Aasheesh Singh 4ebab1a538 minor fixes 2024-01-30 10:23:33 -05:00
Luni-4 b8a1c86962
[xtask] Update xtask publish task (#1189) 2024-01-30 10:05:39 -05:00
Roy Varon f5ac5d8e9f
Added tuple modules (#1186) 2024-01-29 17:56:31 -05:00
Dilshod Tadjibaev 67ffa1e54b
Update documentation and book sections on PyTorch import (#1180) 2024-01-29 17:55:20 -05:00
Aasheesh Singh 61dd2c17b9 1. Address comments :) 
2. Fix formatting
3. remove dead link
 2024-01-29 14:39:24 -05:00
Louis Fortier-Dubois 2defd01342
Add tests: Slice assign vs Cat in LSTM backward (#1146) 
* slice assign test

* added tests but no error

* test for non zero grad

* clippy

* i'm confused

* fix ci
 2024-01-28 21:29:06 -05:00
Chen Wu 868608222b
fix: Batchnorm does not work properly when training on multiple devices (#1167) 2024-01-27 10:07:48 -05:00
Joshua Ferguson 4a70a0f8bc
renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 2024-01-27 10:04:50 -05:00
Sylvain Benner 3814c4c9fb
[xtask] New commands: dependencies and vulnerabilities (#1181) 
* [xtask] Add support for vulnerabilities check in xtask

cargo +nightly xtask vulnerability --help

* [xtask] Add support for dependencies check in xtask

cargo xtask dependencies --help

* Make sure all vulnerabilities checks are called with cargo +nightly

* Fix clippy errors

* Use automatic links in docstrings

* Move run function to the top of the file

* Skip dependencies documenation

* pub -> pub(crate)

* Change return type of Sanitizer::flags to &str

* Move the run functions as an impl function of check types enums

* Remove Type suffix in check type enums

* Fix wrong variable name

* cargo_commande -> cargo_crate

* Improve robustness of is_target_supported and add tests

* Reorganize utils module into a directory
 2024-01-27 08:40:41 -05:00
Aasheesh Singh 3e3382c580 fix typo 2024-01-26 06:10:59 -05:00
Aasheesh Singh 9ef9d27012 5. Update dataset.md to explain Dataset transform usages with examples. 2024-01-26 06:05:01 -05:00
Aasheesh Singh 52b11d9a87 4. Add section on Tensor introduction 2024-01-26 04:16:23 -05:00
Aasheesh Singh a5ae89ebee 1. fix broken link in getting-started.md 
2. Make guide documentation and file structure consistent with examples/guide.
3. Add helpful comments in guide.
 2024-01-25 19:17:28 -05:00
Guillaume Lagrange b9bd42959b
Add vision/mnist dataset (#1176) 2024-01-25 16:16:39 -05:00
Dilshod Tadjibaev 0368409eb3
Add support for loading PyTorch `.pt` (weights/states) files directly to model's record (#1085) 2024-01-25 10:20:09 -05:00
unrenormalizable 4ca3e31601
Fixing jupyter nbs + ONNX IR (#1170) 2024-01-24 13:32:01 -05:00
Guillaume Lagrange e1d873abe2
Add Rust concept notes to Burn Book (#1169) 2024-01-24 12:58:14 -05:00
Joshua Ferguson 3b7d9feede
Elementwise pow op (#1133) 2024-01-24 09:46:57 -05:00
Sylvain Benner 707b61051c
Fix doc build on docs.rs (#1168) 2024-01-23 14:13:57 -05:00
Nathaniel Simard eaa4dc3207
Feat/recorder/custom device (#1165) 2024-01-23 13:05:41 -05:00
Guillaume Lagrange e9d1656687
Add model saving and loading examples (#1164) 2024-01-23 12:00:04 -05:00
Nathaniel Simard c9cc1321b6
Update wgpu version (#1166) 2024-01-23 09:50:51 -05:00
Nathaniel Simard 49e49c9bac
Support more elemwise kernel fusion (#1159) 2024-01-22 16:19:20 -05:00
Nathaniel Simard 635cc0a3e7
[burn-fusion] overlapping optimization builders (#1162) 2024-01-22 16:18:59 -05:00
Guillaume Lagrange 191877f14a
New default recorder (#1161) 
* Change DefaultFileRecorder to NamedMpkFileRecorder (no compression)

* Actually, safetensors does not have any checksum for data validation

* Update checksum explainer/recommandation
 2024-01-22 12:50:42 -05:00
Guillaume Lagrange 21772a2662
Fix FullPrecisionSettings type for integers (#1163) 2024-01-22 12:08:56 -05:00
James MacGlashan a752cc172c
Allow raw tensors to be serialized/deserialized directly with serde (#1041) 2024-01-20 11:25:49 -05:00
Sylvain Benner 9063701b53
Remove annoying cargo warning (#1108) 
Package level profiles are ignored when workspace is enabled so
these lines are useless.

This commit removes them completely as panic, lto and rpath cannot be overridden from the workspace root.

Source: https://doc.rust-lang.org/cargo/reference/profiles.html#overrides

```
The precedence for which value is used is done in the following order (first match wins):

[profile.dev.package.name] — A named package.
[profile.dev.package."*"] — For any non-workspace member.
[profile.dev.build-override] — Only for build scripts, proc macros, and their dependencies.
[profile.dev] — Settings in Cargo.toml.
Default values built-in to Cargo.
Overrides cannot specify the panic, lto, or rpath settings.
```
 2024-01-20 09:09:20 -05:00
Aasheesh Singh e3355524a2
Example/regression (#1150) 2024-01-19 21:28:38 -05:00
Nathaniel Simard ebb2306a6b
[burn-fusion] support multiple concurrent optimization streams (#1149) 2024-01-18 13:15:43 -05:00