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

382 Commits

Author SHA1 Message Date
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
Dilshod Tadjibaev 834c7ecc1f
Clean up cargo descriptions and formatting (#403) 2023-06-15 09:20:53 -04:00
Nathaniel Simard 71d7ebbb21
Fix concat backward with more than 1 dim (#402) 2023-06-15 09:18:15 -04:00
Dilshod Tadjibaev d57ca96695
Upgrade dep versions (#399) 2023-06-14 09:55:19 -04:00
Dilshod Tadjibaev 80d145d629
Add run-before-pr.sh script and fix clippy errors (#397) 2023-06-14 09:10:24 -04:00
Dilshod Tadjibaev e2adb294e9
Add comments and clean up readme with examples (#396) 2023-06-14 09:06:45 -04:00
Nathaniel Simard 8ed0775dfd
Perf/wgpu/context (#395) 2023-06-12 14:24:19 -04:00
Louis Fortier-Dubois fd88398ce4
feat/Xavier glorot (#394) 2023-06-11 09:54:44 -04:00
Mathias Insley f8cd38c071
Feat/gru (#393) 2023-06-09 18:56:40 -04:00
Nathaniel Simard 8c9802c363
Feat/wgpu/reduction (#392) 2023-06-08 16:54:36 -04:00
uint256_t dc5aa41ea5
ONNX: Add `Bool` element type (#389) 2023-06-08 16:48:51 -04:00
Nathaniel Simard cb4f049eae
fix: improve assert approx (#390) 2023-06-06 14:34:20 -04:00
Mathias Insley 8a88a868ee
Feat/lstm (#370) 2023-06-06 14:33:22 -04:00
Nathaniel Simard bff752b1a8
Feat/wgpu/index (#386) 2023-06-06 12:29:14 -04:00
Nathaniel Simard c1e1e38a79
Add more ops (#387) 2023-06-06 12:21:20 -04:00
Nathaniel Simard ecc67c58f9
Feat/wgpu/swap dims (#381) 2023-06-04 19:34:35 -04:00
Yu Sun 105c259d44
feat(learner): add RegressionOutput (#380) 2023-06-04 10:21:29 -04:00
Nathaniel Simard 0a205a3603
feat: add basic matmul for wgpu backend (#379) 2023-06-03 10:32:57 -04:00
Yu Sun 498d163a7b
Feat: mse loss (#378) 2023-06-03 10:31:12 -04:00
Nathaniel Simard 974fdfaba1
Feat/wgpu backend setup (#376) 2023-06-02 11:52:47 -04:00
Dilshod Tadjibaev 483f9acca5
Enhance SqliteDatasetWriter to remove tmp on SIGTERM (#375) 
Temporary files were dangling when a process is terminated by SIGINT (ctrl/cmd+c) or SIGTERM. This change used gix-tempfile to wrap a temporary file with a special signal handle to will remove the tmp file when writer is dropped or when program is terminated early.
 2023-06-02 09:49:48 -04:00
Dilshod Tadjibaev f24f91c651
Fix clippy error (#377) 2023-06-01 17:11:20 -04:00
Nathaniel Simard 3b6ba1bbab
Refactor/dim inference (#369) 2023-06-01 12:02:12 -04:00
Dilshod Tadjibaev f05cf7bac3
Add SqliteDatasetWriter to store items in sqlite db (#364) 2023-06-01 09:54:50 -04:00
Dilshod Tadjibaev befec6bcb2
Dataset audio: add Copy derive (#372) 2023-05-31 10:57:30 -04:00