Commit Graph

426 Commits

Author SHA1 Message Date
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
Nathaniel Simard 513b9281c2
Feat/matmul/faster (#479) 2023-07-07 12:00:37 -04:00
Nathaniel Simard 261aa952c0
Add data benchmark (#474) 2023-07-07 10:21:24 -04:00
Nathaniel Simard 017485b9ea
Refactor/wgpu/ops (#472) 2023-07-06 14:26:27 -04:00
Nathaniel Simard 04ad14a32a
refactor: wgpu reductions (#471) 2023-07-06 11:40:37 -04:00
Dilshod Tadjibaev d78f25f922
Add WebGPU backend to the feature list (#470) 2023-07-06 08:51:57 -04:00
Dilshod Tadjibaev e62ee1269b
Fix burn-tch's random implementation for standard dist (#469) 2023-07-06 08:50:50 -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 65bf6c1cbb
Refactor index => slice (#466) 2023-07-05 16:30:11 -04:00
Louis Fortier-Dubois 042d2201d2
Feat/wgpu/matmul benches (#465) 2023-07-04 09:34:42 -04:00
Nathaniel Simard 98d4abd9c0
Perf/wgpu/comparison (#460) 2023-07-04 09:33:05 -04:00
Dilshod Tadjibaev a2b2d99aa3
Add Tensor as a module constant (#459) 2023-07-03 13:00:18 -04:00
Louis Fortier-Dubois 57ae13c1c0
Perf/wgpu/matmul tests (#458) 2023-07-02 16:09:24 -04:00
Nathaniel Simard 695d30918f
fix: wgpu cat can support bigger tensors (#456) 2023-07-02 12:59:32 -04:00
Nathaniel Simard a1c9970373
Refactor/wgpu/mask (#453) 2023-07-02 12:59:22 -04:00
Nathaniel Simard 5341c0260b
Perf/wgpu/benches (#448) 2023-07-02 12:57:15 -04:00
Dilshod Tadjibaev aced634827
Remove hardcoded "1" stride from Conv1d (#451) 2023-07-01 10:31:16 -04:00
Nathaniel Simard 86b23d5117
fix: training epoch progress (#450) 2023-07-01 10:30:23 -04:00
Nathaniel Simard 77219f7262
Perf/unary binary (#446) 2023-06-29 08:56:35 -04:00
Louis Fortier-Dubois f99fe0fadd
Matmul 2D Tiling (#442) 2023-06-28 16:48:15 -04:00
Nathaniel Simard f42176e93a
chore: upgrade syn (#445) 2023-06-28 12:50:28 -04:00
Nathaniel Simard 13836ac599
Fix: wgpu powf (#443) 2023-06-28 12:50:16 -04:00
Dilshod Tadjibaev 4c94679a8e
Add supported platforms section to readme (#444) 2023-06-28 09:54:10 -04:00
Nathaniel Simard dc216f574b
fix ndarray index select assign (#441) 2023-06-27 12:48:01 -04:00
Nathaniel Simard d26dead299
fix arange (#440) 2023-06-27 12:47:49 -04:00
Dilshod Tadjibaev 85521a1837
Minor stylistic changes to the PR tempalte (#437) 2023-06-27 08:02:59 -04:00
Nathaniel Simard 4ab3ab6554
Feat/wgpu/gather scatter (#436) 2023-06-27 08:02:47 -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
Dilshod Tadjibaev 7f5f0883ae
Add PR template and fix script name (#429) 2023-06-26 15:58:37 -04:00
Dilshod Tadjibaev 79e33109ee
Create dependabot.yml (#431) 2023-06-26 15:57:56 -04:00
Nathaniel Simard b6be22b855
Simpler CI (#407) 2023-06-26 08:58:54 -04:00