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

476 Commits

Author SHA1 Message Date
Ikko Eltociear Ashimine 9361193b5d
Fix typo in tensor.rs (#599) 
overriden -> overridden
 2023-08-07 12:28:08 -04:00
Gadersd ed255c5561
Use buffered io for massive performance gains when loading and saving… (#593) 2023-08-06 12:56:27 -04:00
Nathaniel Simard 3d9f1894ae
Link models repository (#596) 2023-08-06 12:53:33 -04:00
mashirooooo 72f6bf3ca7
feat: check python venv environment initialized (#588) 2023-08-06 11:49:35 -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
Nathaniel Simard ce8a175aa4
Feat/conv transpose1d backward (#586) 2023-08-06 10:50:10 -04:00
Dilshod Tadjibaev 1554a3c898
Full support for ONNX scalar operators and Constants (#578) 2023-08-04 16:51:51 -04:00
Nathaniel Simard ca9a8808d9
Feat/adaptive avg pool1d (#585) 2023-08-04 13:55:18 -04:00
Nathaniel Simard 8436d4ff66
Feat/tensor/adaptive avg pool2d (#572) 2023-08-04 10:23:59 -04:00
Nathaniel Simard 597eab524d
Feat/conv transpose2d (#574) 2023-08-03 15:42:18 -04:00
Chris Cowan 706a6ea99e
Fix huggingface downloader Windows support (#564) (#575) 2023-08-03 09:23:12 -04:00
dependabot[bot] b7ad23bd87
Update serial_test requirement from 0.5.0 to 2.0.0 (#579) 
Updates the requirements on [serial_test](https://github.com/palfrey/serial_test) to permit the latest version.
- [Release notes](https://github.com/palfrey/serial_test/releases)
- [Commits](https://github.com/palfrey/serial_test/compare/v0.5.0...v2.0.0)

---
updated-dependencies:
- dependency-name: serial_test
  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-03 09:21:47 -04:00
Louis Fortier-Dubois d5f9f69cea
Refactor/wgpu/prng (#576) 2023-08-02 16:08:50 -04:00
mmalczak 73fb0eaa7e
Addition of abs tensor opperator #506 (#553) 2023-08-01 18:25:14 -04:00
Louis Fortier-Dubois 87125da6c9
Feat/wgpu/prng bernoulli (#571) 2023-08-01 12:54:22 -04:00
Louis Fortier-Dubois a69788ad4b
Feat/wgpu/prng normal (#570) 2023-08-01 10:59:49 -04:00
Louis Fortier-Dubois daedec6f6d
Feat/wgpu/prng (#560) 
* wip

* wip

* default prng working but not perfectly random

* format

* format

* fix prng algo

* clippy

* uniform prng

* refactor tests for prng

* cleanup
 2023-08-01 09:26:49 -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
Dilshod Tadjibaev f453f47efb
ONNX Concat op (#561) 2023-07-31 15:46:37 -04:00
Nathaniel Simard e10cf4adec
Update graphics drivers (#568) 2023-07-31 12:55:13 -04:00
Dilshod Tadjibaev 6b459fde82
Update Conv2d source generation to support all attributes (#558) 2023-07-28 12:52:23 -04:00
Luni-4 7dbd43d111
Rewrite `publish.sh` script in Rust language (#556) 2023-07-28 12:51:41 -04:00
Louis Fortier-Dubois aa4af29e3f
Matmul speedup (contiguous load) (#559) 2023-07-28 10:41:27 -04:00
Luni-4 67ce5ec62c
Implement binary operators for `burn-import` (#532) 2023-07-27 19:37:10 -04:00
Luni-4 2148ac43a4
burn-import: Add transpose operator (#546) 2023-07-27 19:34:43 -04:00
Nathaniel Simard f0a71350cf
refactor: burn-import unary operators (#548) 2023-07-27 13:20:11 -04:00
Dilshod Tadjibaev 74c41bdda2
Add clamp, clamp_min, clamp_max tensor ops (#550) 2023-07-26 20:02:38 -04:00
Luni-4 8e1e29b599
Refactor test code and some missing operators test in `burn-import` (#547) 2023-07-26 14:24:08 -04:00
Luni-4 2af347ca0d
Add `Reshape` operator to `burn-import` (#545) 2023-07-26 14:23:01 -04:00
Tuhin Tarafder 6c2c7e79c1
Add functions to cast tensor to different kinds (#530) 2023-07-26 14:20:37 -04:00
Nathaniel Simard 0a5a2d729a
chore: bump version for next release (#533) 2023-07-26 09:46:28 -04:00
nathaniel 7c9852b0a1 CI: add wgpu to publish workflow 2023-07-25 10:54:58 -04:00
Louis Fortier-Dubois 589b4503df
add wgpu readme (#531) 2023-07-25 10:44:53 -04:00
Louis Fortier-Dubois 6a25eb1461
Refactor/example/text classification (#527) 2023-07-25 09:50:00 -04:00
Luni-4 914820e6e4
burn-core: Use a specific structure for 1d padding (#529) 2023-07-24 19:31:37 -04:00
Will Brickner e387977eee
Add a `transpose` sugar: `tensor^T` (#528) 2023-07-24 19:30:58 -04:00
Louis Fortier-Dubois 7154bde53a
patch tanh bug on mac os (#520) 2023-07-24 19:29:10 -04:00
polina guseva 64090a582b
Add static full method for tensor intialization with custom values (#486) 
---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2023-07-24 13:03:27 -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
Luni-4 99af7c900f
burn-import: Implement `Add` operator for onnx (#524) 2023-07-24 12:56:53 -04:00
Luni-4 e066d95d2e
Implement padding for conv2d (#523) 2023-07-24 12:54:45 -04:00
Dilshod Tadjibaev b83203bc1c
Add disk clean up step for CI tests (#525) 2023-07-24 12:18:54 -04:00
Nathaniel Simard eaef215b17
Feat: wgpu cast tensor type (#515) 2023-07-24 11:50:44 -04:00
Luni-4 f70ff4dc54
Add MaxPool2d to burn-import (#507) 2023-07-23 19:28:36 -04:00
Louis Fortier-Dubois 9aca1837c2
Example/wgpu/mnist (#514) 
* add wgpu for mnist

* auto graphics api

* fix display tests

* clipy
 2023-07-20 17:12:13 -04:00
Nathaniel Simard d7ce52f0da
Feat/wgpu/conv (#512) 2023-07-20 15:14:42 -04:00
Nathaniel Simard 3a153d5bd0
Feat/wgpu/default device (#513) 2023-07-20 15:07:50 -04:00
Louis Fortier-Dubois df2f1492f8
Feat/wgpu/matmul transpose (#509) 2023-07-20 14:21:58 -04:00
Louis Fortier-Dubois 4b60c0e7a0
continuous to contiguous (#511) 2023-07-20 11:28:35 -04:00
Luni-4 62ab554df8
burn-import: Add a list of all ONNX operators (#508) 2023-07-19 15:07:56 -04:00