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

209 Commits

Author SHA1 Message Date
Nathaniel Simard e6e7f4de42
feat: inplace tensor api. (#187) 2023-03-01 10:55:51 -05:00
Nathaniel Simard 25deb5a13b
Refactor/autodiff (#186) 2023-02-28 20:01:26 -05:00
Dilshod Tadjibaev a62738b0f4
Add a workaround script for arm64 tch-rs build issue (#180) (#182) 
* Add a workaround script for arm64 tch-rs build issue (#180)
 2023-02-28 20:01:01 -05:00
Dilshod Tadjibaev 37806b576c
Fix FailToDownloadPythonDependencies error (#82) (#185) 2023-02-28 09:23:49 -05:00
Dilshod Tadjibaev c510e8bfb5
Add accelerate framework support for burn-ndarray (#183) (#184) 2023-02-28 09:09:47 -05:00
Dilshod Tadjibaev 9df3b6a3be
Add new packages and tests with no-std options in CI (#168) (#179) 2023-02-26 14:18:13 -05:00
Dilshod Tadjibaev b7be9cec15
Fix CI template that was causing error (#168) (#178) 2023-02-26 13:00:56 -05:00
Dilshod Tadjibaev 83e43a3eac
Add no-std steps to the test template (#168) (#177) 2023-02-26 12:19:47 -05:00
Nathaniel Simard 6ba9de2868
perf: remove unecessary masking in dropout (#172) 2023-02-26 11:43:46 -05:00
Dilshod Tadjibaev fb925acc73
Make burn and burn-core packages no_std compatible (#168) (#173) 
* Make burn-ndarray and burn-tensor no_std compatible (#168)
 2023-02-25 09:38:01 -05:00
Nathaniel Simard 0d50ed0aea
perf: ndarray features (#170) 2023-02-22 08:44:20 -05:00
Dilshod Tadjibaev 9091363ada
Make burn-ndarray and burn-tensor no_std compatible (#168) (#169) 2023-02-21 08:35:24 -05:00
Nathaniel Simard 23a58acfdb
fix: arange poor performance (#167) 2023-02-18 15:56:32 -05:00
Nathaniel Simard e4e9a5dcca
fix(burn-tch): tensor comparison ops inplace don't have the right tch kind (#166) 2023-02-18 13:59:49 -05:00
Nathaniel Simard 7d2f43dfca
Refactor Tensor API (#163) 2023-02-17 17:31:20 -05:00
nathaniel 54e2c1a600 chore: add doc issue template 2023-02-16 09:09:53 -05:00
Aaron Roney f907d0025d
Add feature to load config / state from binary slice. (#164) 2023-02-16 08:43:43 -05:00
Nathaniel Simard 2401d8ad96
Prepare next release (#161) 2023-02-12 15:32:29 -05:00
nathaniel 2c4fc72c47 Update README 2023-02-12 15:17:30 -05:00
Yu Sun 0b85cb0eed
feat(trait-TensorOps): add log1p (#160) 2023-02-11 13:30:50 -05:00
Nathaniel Simard 993a6e3095
Feat/batch norm (#158) 2023-02-02 17:22:40 -05:00
Nathaniel Simard c7963d8485
refactor: device functions (#157) 2023-01-27 18:37:21 -05:00
Nathaniel Simard 2d4e514b41
Refactor/shape function (#156) 2023-01-27 15:18:55 -05:00
Makro f6f0d0e4f3
Add cos, sin and tanh operations (#155) 
* Add cos, sin and tanh operations

* Add tests

* Fix formatting
 2023-01-24 19:40:30 -05:00
Nathaniel Simard b7aa066e5c
Feat/nn max pool2d (#153) 2023-01-21 17:06:39 -05:00
Nathaniel Simard 34d233cd3e
Feat/max pooling backend (#152) 2023-01-21 15:39:21 -05:00
Nathaniel Simard 1d0d92a269
Feat/nn conv (#148) 2023-01-12 20:22:48 -05:00
Nathaniel Simard 745c88f0a0
Feat/conv (#147) 2023-01-11 18:33:09 -05:00
nathaniel cadf494f1e Update README.md 2023-01-02 10:40:30 -05:00
Nathaniel Simard 54c30abfb8
Refactor/burn core (#143) 2023-01-01 19:21:08 -05:00
Nathaniel Simard 2f179f12c9
Bump versions (#141) 2022-12-30 15:15:51 -05:00
Nathaniel Simard eea5a263bf
Feat/adam optimizer (#140) 2022-12-30 15:02:43 -05:00
Nathaniel Simard 248039da0a
Refactor/metric adaptor (#139) 2022-12-26 16:30:25 -05:00
Visual 567adfb93e
refactor: fix all clippy warnings (#137) 2022-12-25 11:22:25 -05:00
Visual 85f98b9d54
refactor, feat: clean Cargo.toml files, upgrade tch to 0.10 (#131) 
* Clean Cargo.toml files, upgrade tch to 0.10

* Add pull_request hook to test.yml workflow
 2022-12-25 10:36:23 -05:00
Nathaniel Simard 1ec35a9e1b
feat: from floats (#87) 2022-12-25 10:11:20 -05:00
Nathaniel Simard 3a9dfe6097
feat: cross entropy loss (#130) 2022-12-25 10:10:22 -05:00
Nathaniel Simard 1a1d86dc3e
refactor: save and load state (#129) 2022-12-24 13:02:37 -05:00
Nathaniel Simard 3a91c2c48e
Feat/multi device (#128) 2022-12-20 18:01:58 -05:00
Nathaniel Simard a599eaed88
Feat/module visitor (#127) 2022-12-17 14:25:36 -05:00
Nathaniel Simard d9592411c2
Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00
Nathaniel Simard 63d8d39517
Feat/autoregressive transformer (#125) 2022-12-10 15:47:57 -05:00
Nathaniel Simard b99b23e1a7
refactor: backends (#124) 2022-12-02 19:28:34 -05:00
Nathaniel Simard eee90a5c9e
Example/text classification (#123) 2022-12-02 17:42:49 -05:00
Nathaniel Simard 7c38a980c1
feat: improve bool tensor (#122) 2022-12-01 19:43:36 -05:00
Jakub Lewandowski cdc29a02d2
Change links in README to work on crates.io (#114) 2022-11-28 20:08:36 -05:00
Nathaniel Simard 9cade192fd
Feat/transformer encoder (#119) 2022-11-26 20:06:43 -05:00
Nathaniel Simard 8bd0b17296
Feat/mha (#118) 2022-11-26 15:48:26 -05:00
Nathaniel Simard 46d06f0c90
feat: module init (#117) 2022-11-25 22:02:26 -05:00
Nathaniel Simard 4abc281dae
Refactor/forward pass (#116) 2022-11-25 21:14:44 -05:00