good first issue
Métriques du dépôt
- Stars
- (336 stars)
- Métriques de merge PR
- (Métriques PR en attente)
Description
NNlib
- activations
- tanh_fast
- sigmoid
- sigmoid_fast
- softmax!
- logsoftmax!
- conv / conv! (#258)
- maxpool / maxpool! (#258)
- meanpool / meanpool! (#258)
- batched_mul / batched_mul! (#258)
- padding
- pad_constant
- pad_circular
- pad_repeat
- pad_zeros
- pad_reflect
- pad_symmetric
- ∇conv_data / ∇conv_data!
- ∇conv_filter / ∇conv_filter!
- gather / gather!
- scatter / scatter! https://github.com/EnzymeAD/Reactant.jl/pull/1395
-
dot_product_attention -
pixel_shuffle - batchnorm (has a special stablehlo impl) - this needs to wait for a corresponding adjoint on EnzymeJAX end
- grid_sample / grid_sample!
- upsampling
-
nearest - linear
- bilinear
- trilinear
-
- fold / fold!
- unfold / unfold!
Some of these might not even be needed, but we should test the generated IR. Strikethrough denotes we don't need specialized handling for these operations
Feel free to add missing operations to the list