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

506 Commits

Author SHA1 Message Date
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
Nathaniel Simard 6c834d3b22
Feat/wgpu/index select (#425) 2023-06-23 09:31:49 -04:00
Dilshod Tadjibaev f0b266c8a3
Add missing docs for burn-wgpu and burn-tch (#427) 2023-06-23 09:31:37 -04:00
Dilshod Tadjibaev 825aaa9977
Add missing documents (#424) 2023-06-23 09:28:34 -04:00
Dilshod Tadjibaev eda241f8cf
Add missing docs and enable missing_docs warn lint (#420) 2023-06-21 14:12:13 -04:00
Nathaniel Simard c4e4c25fef
Feat/wgpu/mask (#422) 2023-06-21 13:01:18 -04:00
Yu Sun 9c5afa3469
fix(loss): Declare all contents of reduction as public (#423) 2023-06-21 13:00:57 -04:00
Dilshod Tadjibaev fce45f51be
Doc fixes (#418) 2023-06-21 12:32:50 -04:00
Dilshod Tadjibaev 73a88d8209
Remove class grouping for speech-commands dataset (#419) 
Instead of grouping the classes when generating the data, we are letting the users to decide how to group the data.
 2023-06-21 11:35:56 -04:00
Dilshod Tadjibaev bd42076d3f
Clean up the main readme to pop a bit (#421) 2023-06-21 10:24:29 -04:00
Nathaniel Simard 3b9c997513
feat: provide embedding default impl (#417) 2023-06-20 16:16:31 -04:00
Nathaniel Simard b6b684b9f2
Feat/wgpu/cat (#416) 2023-06-20 16:16:10 -04:00
Dilshod Tadjibaev 4683acf726
Minor clean up of doc formatting and remove outdated TODO (#415) 2023-06-20 10:05:28 -04:00
Nathaniel Simard 4d40bde7b9
feat: argmax + argmin (#412) 2023-06-20 10:03:00 -04:00
Nathaniel Simard 323261b594
Feat/wgpu/comparisons (#411) 2023-06-19 19:11:49 -04:00
Louis Fortier-Dubois e5cc50a837
refactor/initializer (#398) 
* add tests to linear forward

* swap linear weight dims

* rename uniformdefault

* wip bias in init

* refactor initializer

* formatting

* fix get(0) to first

* wip refactor

* builder pattern + kaiming init

* kaiming initializer and major initializer refactor

* fix fan out

* easier fixes

* remove initializer options

* clippy fix

* revert to input*output linear

* gru uses xavier init

* should be normal actually

---------

Co-authored-by: louisfd <louisfortier-dubois@Louiss-MacBook-Pro.local>
 2023-06-19 16:55:11 -04:00
Nathaniel Simard 40743c477a
Feat/onnx/more ops (#410) 2023-06-19 16:10:50 -04:00
Nathaniel Simard a8624590af
feat: mask_where (#409) 2023-06-18 15:04:28 -04:00
Dilshod Tadjibaev 11bdceff56
Remove inference testing against dataset (#408) 
This test was redundant and required the dataset to be downloaded.
 2023-06-17 12:59:26 -04:00
Nathaniel Simard b6f88f8fca
refactor: better templating for wgsl shaders (#404) 2023-06-15 11:59:37 -04:00
Nathaniel Simard 18b7d5202d
Fix wgpu context creation (#405) 2023-06-15 11:57:56 -04:00