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

246 Commits

Author SHA1 Message Date
Dilshod Tadjibaev 3264b1007c
Changes and fixes required for squeezenet1.1-opset7 onnx (#639) 2023-08-15 15:37:52 -04:00
Caio Piccirillo cb283a9e5b
Max pool1d (#602) 2023-08-09 16:13:48 -04:00
Caio Piccirillo 1d3bbaab13
Typos (#608) 2023-08-08 17:57:51 -04:00
Nathaniel Simard 441a7011ce
Feat/tensor casting (#604) 2023-08-08 10:02:17 -04:00
Ikko Eltociear Ashimine 9361193b5d
Fix typo in tensor.rs (#599) 
overriden -> overridden
 2023-08-07 12:28:08 -04:00
Nathaniel Simard ce8a175aa4
Feat/conv transpose1d backward (#586) 2023-08-06 10:50:10 -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
mmalczak 73fb0eaa7e
Addition of abs tensor opperator #506 (#553) 2023-08-01 18:25:14 -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
Dilshod Tadjibaev 74c41bdda2
Add clamp, clamp_min, clamp_max tensor ops (#550) 2023-07-26 20:02:38 -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
Louis Fortier-Dubois 589b4503df
add wgpu readme (#531) 2023-07-25 10:44:53 -04:00
Will Brickner e387977eee
Add a `transpose` sugar: `tensor^T` (#528) 2023-07-24 19:30:58 -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
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
Louis Fortier-Dubois 4b60c0e7a0
continuous to contiguous (#511) 2023-07-20 11:28:35 -04:00
Nathaniel Simard c4afff182f
Feat/wgpu/max pool2d (#500) 2023-07-14 13:58:08 -04:00
Dilshod Tadjibaev ef421f0ae9
Add arange with steps op for int tensor (#490) 2023-07-13 17:09:57 -04:00
Nathaniel Simard a2ac2057d8
Fix: wgpu scatter with different shapes (#489) 2023-07-12 13:21:19 -04:00
Jacob cb1ac26bb3
Add SiLU activation function (#482) 2023-07-09 11:28:14 -04:00
Nathaniel Simard 04ad14a32a
refactor: wgpu reductions (#471) 2023-07-06 11:40:37 -04:00
Dilshod Tadjibaev e62ee1269b
Fix burn-tch's random implementation for standard dist (#469) 2023-07-06 08:50:50 -04:00
Nathaniel Simard 65bf6c1cbb
Refactor index => slice (#466) 2023-07-05 16:30:11 -04:00
Nathaniel Simard 98d4abd9c0
Perf/wgpu/comparison (#460) 2023-07-04 09:33:05 -04:00
Nathaniel Simard a1c9970373
Refactor/wgpu/mask (#453) 2023-07-02 12:59:22 -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 13836ac599
Fix: wgpu powf (#443) 2023-06-28 12:50:16 -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 61291cc30c
Revert temp half-rs fix for no_std (#430) 
Fixes #268
 2023-06-26 15:59:06 -04:00
Nathaniel Simard 6c834d3b22
Feat/wgpu/index select (#425) 2023-06-23 09:31:49 -04:00
Dilshod Tadjibaev 825aaa9977
Add missing documents (#424) 2023-06-23 09:28:34 -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
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
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 834c7ecc1f
Clean up cargo descriptions and formatting (#403) 2023-06-15 09:20:53 -04:00
Louis Fortier-Dubois fd88398ce4
feat/Xavier glorot (#394) 2023-06-11 09:54:44 -04:00
Nathaniel Simard 8c9802c363
Feat/wgpu/reduction (#392) 2023-06-08 16:54:36 -04:00