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

507 Commits

Author SHA1 Message Date
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
Louis Fortier-Dubois 57a5476c89
bugfix for macos test (#503) 2023-07-18 16:15:00 -04:00
Louis Fortier-Dubois 5ece894e02
Bugfix/matmul/asymetric shapes (#504) 2023-07-18 16:14:43 -04:00
Nathaniel Simard f7c7d35ef5
Feat/wgpu/avg pooling (#502) 2023-07-18 11:36:57 -04:00
Luni-4 93723205e0
Add Sigmoid operator to burn-import (#501) 2023-07-17 12:45:44 -04:00
Nathaniel Simard c4afff182f
Feat/wgpu/max pool2d (#500) 2023-07-14 13:58:08 -04:00
dengelt 860e4433c6
Install dataset downloader python deps in venv (#496) 2023-07-14 13:22:44 -04:00
Nathaniel Simard 0d17cde1cc
Increase repeat performance in tch backend (#499) 2023-07-14 13:21:56 -04:00
Luni-4 f176f67ab6
Move main file into the bin directory for onnx_inference example (#498) 2023-07-14 10:14:37 -04:00
Dilshod Tadjibaev f433292a3f
Fix intermittent test failure (#497) 2023-07-13 19:40:54 -04:00
Luni-4 ffdb0feb29
Hide crate name in onnx-import crate and onnx-inference example (#493) 2023-07-13 19:39:17 -04:00
Dilshod Tadjibaev e267fc1e6f
Temporarily disable broken tests for M1 Mac (#491) 
A temp workaround for #480
 2023-07-13 17:10:26 -04:00
Dilshod Tadjibaev ef421f0ae9
Add arange with steps op for int tensor (#490) 2023-07-13 17:09:57 -04:00
Dilshod Tadjibaev 53c088209d
Fix new clippy warnings that cause the CI to fail (#494) 2023-07-13 13:39:39 -04:00
Nathaniel Simard a2ac2057d8
Fix: wgpu scatter with different shapes (#489) 2023-07-12 13:21:19 -04:00
Nathaniel Simard 4bc740a864
Fix: can't use init or load with a record with correct gradient requi… (#488) 2023-07-11 15:01:14 -04:00
Dilshod Tadjibaev 20213ade73
Add sinusoidal positional embedding module (#481) 2023-07-11 10:55:29 -04:00
Jacob cb1ac26bb3
Add SiLU activation function (#482) 2023-07-09 11:28:14 -04:00
Dilshod Tadjibaev 95ea764a33
Update README.md to fix a broken link (#484) 
Fixes #483
 2023-07-09 11:27:41 -04:00
Nathaniel Simard ddbbe39d74
Fix: wgpu cat + copy ops (#477) 2023-07-07 12:19:10 -04:00