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

778 Commits

Author SHA1 Message Date
louisfd 1fc59b0861 minor clean 2023-12-21 16:12:35 -05:00
louisfd bb18452136 Merge branch 'main' into fix/devauto_tests 2023-12-21 15:53:24 -05:00
louisfd 8acf4b223b remove from_devauto occurrences 2023-12-21 15:53:04 -05:00
Louis Fortier-Dubois b070706310
Fix/wgpu/tanh (#1090) 2023-12-21 14:12:49 -05:00
louisfd 5a5ea06db2 wip removing devauto 2023-12-21 13:51:14 -05:00
Nathaniel Simard d82e6b157b
Fix tests (#1089) 
* Fix tests

* Fix fmt

* Fix CI
 2023-12-21 13:06:19 -05:00
Kirill Mavreshko 1fd07fcb4a
Explicit device tensors (#1081) 2023-12-20 17:49:59 -05:00
Kelvin Wu 7c6f017c98
Implement chunk for different backends (#1032) 2023-12-20 13:35:59 -05:00
pacowong c1cb77ac2e
Add mish and softplus activation function (#1071) 2023-12-20 12:41:23 -05:00
Louis Fortier-Dubois 4fb5b74087
fix multithread progress aggregation in dataloader (#1083) 2023-12-19 16:28:57 -05:00
Nathaniel Simard 75062c51e0
Refactor/wgpu/binary (#1078) 
* Refactor binary

* Fix

* Oups

* Preparation for cmp

* Refactor comparison

* Remove templates

* Cleanup

* Cleanup

* Fix typo

* Code review
 2023-12-19 14:44:55 -05:00
Nathaniel Simard b5c49c5bf7
Fusion wgpu compilation cache (#1069) 
* Refactor fusion in the wgpu backend

* WIP

* Refactor

* WIP

* Fix inplace ops

* Works ish

* Cleanup Output

* Refactoring

* Refactor Clamp

* Cleanup

* Cleanup

* Updates

* Fix CI

* Code review
 2023-12-18 12:16:08 -05:00
Nathaniel Simard 042454a9db
Example: Update Text Classification (#1044) 
* Example: Update Text Classification

* cargo fmt

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-12-18 08:42:07 -05:00
Louis Fortier-Dubois 3f06e81f2d
Add stride field to reduce key (#1070) 2023-12-15 13:35:46 -05:00
Luni-4 467e65247b ci/Add Windows pipeline (#925) 
Co-Authored-By: Sylvain Benner <sylvain.benner@gmail.com>
 2023-12-14 18:48:59 -05:00
AuruTus fc97a28f16
fix/ndarray: remove reversed axes check (#1058) 2023-12-14 18:09:24 -05:00
Louis Fortier-Dubois 4608cd97d2
Feat/wgpu/repeat (#1068) 2023-12-14 18:08:37 -05:00
Nathaniel Simard 2ca1fd5a92
Fix bugs (#1064) 2023-12-12 16:19:01 -05:00
Jantxu 70eda6b34e
use web-time in case of wasm (#1060) 2023-12-12 13:47:10 -05:00
Alex Errant 610d64095e
cargo +nightly fmt (#1017) 2023-12-12 13:29:06 -05:00
Luni-4 1a5f252ac6
Ci/Improve scheduled tasks (#1028) 2023-12-12 09:42:33 -05:00
David Chavez 71d3c1d142
chore(infra): Share some properties across workspace (#1039) 2023-12-12 09:39:07 -05:00
David Chavez 3066196297
Fix where sqrt should’ve been powf(2.) (#1052) 2023-12-06 19:02:15 -05:00
David Chavez ef09b637b9
chore(web): better panic (#1051) 2023-12-05 07:30:29 -05:00
David Chavez d9f93d31b4
test burn-ndarray also with accelerate (#1050) 2023-12-04 13:19:46 -05:00
David Chavez 82f5722ca1
chore(ci): CI grouping and refactoring (#1024) 2023-12-04 11:14:50 -05:00
Louis Fortier-Dubois 8fc52113bc
Chore/bump v12 (#1048) 2023-12-04 10:47:54 -05:00
Louis Fortier-Dubois 3088c466a5
patch 0.11.1 (#1047) 2023-12-04 10:18:30 -05:00
Nathaniel Simard 0b16786729
Fix/orphan cleanup (#1043) 2023-12-04 09:44:17 -05:00
Nathaniel Simard 9d92f9a7b8
Update readme (#1045) 2023-12-04 09:39:55 -05:00
Timothy 52811f9938
Fix some issues in Burn book (#1042) 2023-12-04 09:23:47 -05:00
David Chavez cb8d4dac62
Fix building docs (#1040) 2023-12-02 12:08:28 -05:00
nathaniel 3a8dcfb5f4 Fix burn-train feature flag 2023-12-01 16:01:22 -05:00
Louis Fortier-Dubois 4d7944912d
fix gatter typo (#1036) 2023-12-01 15:28:25 -05:00
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Dilshod Tadjibaev 4192490b88
Update with demo links (#1033) 2023-12-01 13:07:13 -05:00
Nathaniel Simard 12911c3ad6
Fix/mnist inference web (#1029) 2023-12-01 12:05:31 -05:00
Nathaniel Simard 670280dda2
Feat/fusion/cache (#1020) 2023-12-01 12:05:11 -05:00
Nathaniel Simard b0de56da29
Fix double broadcast with tch (#1026) 
* Fix double broadcast with tch

* More fixes

* Fix clippy warm
 2023-12-01 10:02:57 -05:00
Nathaniel Simard f6d14f1b1a
Refactor feature flags (#1025) 2023-12-01 09:48:28 -05:00
wcshds 1f18cf4adf
implement DoubleEndedIterator trait for DimIter (#1022) 
* implement DoubleEndedIterator for DimIter

* return early in next() and next_back()

* return early in next() and next_back()

* remove #[rustfmt::skip]
 2023-11-30 12:26:16 -06:00
Dilshod Tadjibaev 5e4e0b6bb3
Change ONNX default record type to NamedMpk (#1019) 
* Change ONNX default record type to NamedMpk

This temporarily workaround of #952 bug.

* Fix formatting

* Fix URL in the doc
 2023-11-30 13:22:34 -05:00
Dilshod Tadjibaev 947bb60fff
Add ConvTranspose2d ONNX OP (#1018) 
* [~] burn-import: ConvTranspose initial support added.

* Complete conv transpose2d ONNX op

* Fix tests

* Changes per PR feedback

---------

Co-authored-by: Никита Пацакула <pacakula@dml001.kribrum.ru>
 2023-11-30 12:22:06 -06:00
Luni-4 ba1de9c654
Ci/Schedule dependencies checks (#1014) 
* ci: Add audit file

* ci: Schedule dependencies checks

* mnist-inference-web: Remove unused dep

* Run when a new release is created

* Fix comment
 2023-11-30 13:06:27 -05:00
David Chavez f73136e3df
chore(candle): Allow enabling accelerate (#1009) 
* chore(candle): Allow enabling accelerate

* Temporarily disable test for accelerate feature

* Allow enabling accelerate from upstream

* Update the README

* Have xtask also test using accelerate

* Renable failing test

* Fix matmul on candle when using accelerate

* Add additional comment to xtask method
 2023-11-30 13:03:00 -05:00
Luni-4 1d4e91ad32
ci/Vulnerabilities checker (#964) 
* ci: Add vulnerabilities checker

* Fix

* Fix

* Fix

* More options

* Schedule vulnerabilities

* Remove
 2023-11-30 13:01:34 -05:00
Will Brickner 03af140e12
Implement Quiet Softmax (`Attention Is Off By One`) (#692) 
* Added quiet_softmax

* Undid bad formatting

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2023-11-30 12:58:30 -05:00
David Chavez f0c75aa748
chore(ci): run workflow on macOS (#1010) 2023-11-29 16:19:11 -05:00
Dilshod Tadjibaev fed618b265
Add Cos, Exp, Gelu, Log and Neg unary ONNX ops (#1013) 
* Add ONNX Cos OP

* Add ONNX Exp OP

* Add ONNX Gelu OP

* Add ONNX Log OP

* Allow approx_constant clippy rule in generated model code

* Add ONNX Neg OP

* Fix tests with custom imports for unary nodes

* Add scalar tests for Sqrt
 2023-11-29 14:57:52 -06:00
Maksym Hazevych 87393b2070
Updated explanation to match code (#1012) 
Updated the explanation at the beginning to match the actual changes to 'Cargo.toml'. Moved down the 'edition' property to match the style of the initial 'Cargo.toml' file. Removed unnecessary comment for a self-explanatory piece of code. Some rewording.
 2023-11-29 09:41:07 -05:00