Repository navigation

All pull requests

Labels (14)Milestones (0)New pull request

fix: return SafeTensorError instead of panic in load_safetensor
#931Infinty-ux opened Jul 1, 2026
Make tests pass on stable + support Pool2D & Conv{1,2}d
#927twitchyliquid64 opened Jul 23, 2024
fix: small typo in advanced-train-loop.rs
#923tristan-f-r opened Mar 2, 2024Contributor
docs(readme): fix examples hyperlink
#922mrtnvgr opened Mar 2, 2024
Avoid CI errors
#920swfsql opened Mar 1, 2024Contributor
Added TryUnstack for tensors.
#919swfsql opened Feb 6, 2024Contributor6 tasks
Add split_tensor_along method
#917swfsql opened Feb 1, 2024Contributor4 tasks
Add RMS normalization
#916swfsql opened Feb 1, 2024Contributor1 task
Add SiLU activation function
#915swfsql opened Feb 1, 2024Contributor
2
Impl core::ops::Sub for Dim types
#914swfsql opened Feb 1, 2024Contributor
Safetensors changes
#913swfsql opened Feb 1, 2024Contributor
Remove functions calling the deprecated _MM_SET_FLUSH_ZERO_MODE
#912rainiwu opened Jan 26, 2024Contributor
3
feat(webgpu): add tensor operations
#904DonIsaac opened Jan 2, 2024Contributor4 of 6 tasks
1
Gradient Clipping
#902swfsql opened Dec 14, 2023Contributor2 tasks
Adds OUTPUT_PADDING to ConvTrans2D
#890swfsql opened Nov 16, 2023Contributor
feat: N-sized tuple (2 <= n <= 8)
#858tristan-f-r opened Aug 26, 2023Contributor
Add complex numbers
#849opfromthestart opened Aug 13, 2023Contributor
7
Adds unstack
#831emchristiansen opened Jul 26, 2023
1
[WIP] Add Serialization for Optimizers and Gradients
#815nkoppel opened Jul 13, 2023Contributor2 of 4 tasks
Mobilenet integration
#794Awpteamoose opened May 30, 2023
1
Mobilenet ops
#793Awpteamoose opened May 25, 2023
Improve allocation caching
#709nkoppel opened Apr 16, 2023Contributor4 of 5 tasks
26
Refactors TryMatmuls for better error messages when dimensions don't match
#681kstavro opened Apr 8, 2023Contributor
5