Skip to content

Pull requests: FluxML/NNlib.jl

New pull request New
26 Open 455 Closed
26 Open 455 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Mooncake rule for NNlib.BatchNorm.
#727 opened Jun 24, 2026 by AstitvaAggarwal Loading…
2 tasks
1
Interface with new Metal primitives
#725 opened Jun 22, 2026 by maleadt Contributor Loading…
6
Add GPU multi-head attention benchmark (NNlib vs LuxLib vs NNkernels)
#721 opened Jun 11, 2026 by CarloLucibello Member Loading…
Add scaled_dot_product_attention with explicit head axis
#719 opened Jun 10, 2026 by CarloLucibello Member • Draft
Test upsample testsuite against multiple eltypes
#716 opened Jun 10, 2026 by CarloLucibello Member Loading…
Remove cuDNN/MIOpen activation broadcast overloads (fixes #504, #509)
#686 opened Jun 7, 2026 by CarloLucibello Member Loading…
3 tasks
3
2
Extending scatter! to work with CUDA sparse arrays (retry of #648)
#672 opened Jan 27, 2026 by alonsoC1s Contributor Loading…
4
Add batched_mul support for oneAPI
#644 opened Aug 17, 2025 by AntonOresten Member Loading…
2 tasks done
2
Fixed the lpnormpool implementation
#639 opened Jun 6, 2025 by IBArbitrary Loading…
4
Add TeLU activation functions telu and telu_fast
#622 opened Jan 7, 2025 by zengmao Loading…
11
Add a switch to turn slow paths into errors
#612 opened Nov 8, 2024 by mcabbott Member Loading…
2 tasks done
1
2
Start replacing tests with DifferentiationInterface
#570 opened Mar 26, 2024 by gdalle • Draft
1 task
3
Possible way to implement a LoopVectorization extension for conv2d & meanpool2d & activations benchmark Run automated benchmarks on CI
#540 opened Sep 26, 2023 by jonas208 Loading…
22
Add EnzymeRules for batchnorm Enzyme
#537 opened Sep 25, 2023 by wsmoses Contributor Loading…
4
[Dummy] Slow down relu intentionally for testing benchmarking tools benchmark Run automated benchmarks on CI
#527 opened Aug 11, 2023 by skyleaworlder Contributor • Draft
2 tasks
3
Using AbstractFloat instead of Float64 in warning check for slow conv
#475 opened Feb 18, 2023 by gabrielpreviato Contributor Loading…
2 tasks done
5
Add norm functions documentation enhancement
#452 opened Jan 3, 2023 by ToucheSir Member Loading…
2 tasks done
26
add dims keyword to gather
#448 opened Dec 26, 2022 by CarloLucibello Member • Draft
4 tasks
permit NNlibCUDA to use Float16 enhancement
#363 opened Nov 16, 2021 by bjarthur Contributor Loading…
5
add topk enhancement
#353 opened Sep 30, 2021 by CarloLucibello Member Loading…
3 tasks
1
9
add groupcount overload for DepthwiseConvDims
#336 opened Jul 17, 2021 by DhairyaLGandhi Member Loading…
2
Add benchmarks, part 1
#304 opened Mar 22, 2021 by dfdx Contributor Loading…
10
Doc corrections and unicode(theta)
#184 opened Mar 7, 2020 by AdarshKumar712 Contributor Loading…
8
Big interface overhaul
#163 opened Jan 31, 2020 by staticfloat Contributor Loading…
1
An attempt to combine dense, depthwise and groupwise conv through DenseConvDims
#146 opened Nov 30, 2019 by arhik Loading…
1 task
5
ProTip! Adding no:label will show everything without a label.