gitextract_0xi0432k/ ├── .buildkite/ │ └── pipeline.yml ├── .codecov.yml ├── .github/ │ ├── copilot-instructions.md │ ├── dependabot.yml │ └── workflows/ │ ├── BenchmarkTrigger.yml │ ├── CompatHelper.yml │ ├── Downstream.yml │ ├── TagBot.yml │ ├── ci.yml │ ├── clean_preview.yml │ └── pr_comment.yml ├── .gitignore ├── LICENSE.md ├── Project.toml ├── README.md ├── benchmark/ │ ├── Project.toml │ ├── benchmarks.jl │ ├── perf_report.jl │ └── runbenchmarks.jl ├── docs/ │ ├── .gitignore │ ├── Project.toml │ ├── make.jl │ └── src/ │ ├── assets/ │ │ ├── flux.css │ │ └── jfk.flac │ ├── audio.md │ ├── index.md │ └── reference.md ├── ext/ │ ├── NNlibAMDGPUExt/ │ │ ├── NNlibAMDGPUExt.jl │ │ ├── activations.jl │ │ ├── batched_mul.jl │ │ ├── conv.jl │ │ └── pool.jl │ ├── NNlibCUDACUDNNExt/ │ │ ├── NNlibCUDACUDNNExt.jl │ │ ├── activations.jl │ │ ├── batchnorm.jl │ │ ├── conv.jl │ │ ├── pooling.jl │ │ └── softmax.jl │ ├── NNlibCUDAExt/ │ │ ├── NNlibCUDAExt.jl │ │ ├── activations.jl │ │ ├── batchedadjtrans.jl │ │ ├── batchedmul.jl │ │ ├── ctc.jl │ │ ├── sampling.jl │ │ ├── scatter.jl │ │ └── utils.jl │ ├── NNlibEnzymeCoreExt/ │ │ └── NNlibEnzymeCoreExt.jl │ ├── NNlibFFTWExt/ │ │ ├── NNlibFFTWExt.jl │ │ └── stft.jl │ ├── NNlibForwardDiffExt.jl │ ├── NNlibMetalExt.jl │ └── NNlibSpecialFunctionsExt.jl ├── src/ │ ├── NNlib.jl │ ├── activations.jl │ ├── attention.jl │ ├── audio/ │ │ ├── mel.jl │ │ ├── spectrogram.jl │ │ └── stft.jl │ ├── batched/ │ │ ├── batchedadjtrans.jl │ │ └── batchedmul.jl │ ├── bias_act.jl │ ├── conv.jl │ ├── conv_bias_act.jl │ ├── ctc.jl │ ├── deprecations.jl │ ├── dim_helpers/ │ │ ├── ConvDims.jl │ │ ├── DenseConvDims.jl │ │ ├── DepthwiseConvDims.jl │ │ └── PoolDims.jl │ ├── dim_helpers.jl │ ├── dropout.jl │ ├── fold.jl │ ├── functions.jl │ ├── gather.jl │ ├── gemm.jl │ ├── impl/ │ │ ├── conv_direct.jl │ │ ├── conv_im2col.jl │ │ ├── depthwiseconv_direct.jl │ │ ├── depthwiseconv_im2col.jl │ │ ├── padding_edges.jl │ │ └── pooling_direct.jl │ ├── normalization.jl │ ├── padding.jl │ ├── pooling.jl │ ├── rotation.jl │ ├── sampling.jl │ ├── scatter.jl │ ├── softmax.jl │ ├── upsample.jl │ └── utils.jl └── test/ ├── Project.toml ├── activations.jl ├── attention.jl ├── batchedmul.jl ├── bias_act.jl ├── conv.jl ├── conv_bias_act.jl ├── ctc.jl ├── dropout.jl ├── ext_amdgpu/ │ ├── activations.jl │ ├── attention.jl │ ├── batched_mul.jl │ ├── batched_repr.jl │ ├── conv.jl │ ├── dropout.jl │ ├── pool.jl │ ├── runtests.jl │ ├── softmax.jl │ └── storage_type.jl ├── ext_cuda/ │ ├── activations.jl │ ├── batchedadjtrans.jl │ ├── batchedmul.jl │ ├── batchnorm.jl │ ├── conv.jl │ ├── ctc.jl │ ├── dropout.jl │ ├── fold.jl │ ├── gather.jl │ ├── pooling.jl │ ├── runtests.jl │ ├── sampling.jl │ ├── scatter.jl │ ├── softmax.jl │ └── test_utils.jl ├── ext_metal/ │ ├── activations.jl │ └── runtests.jl ├── functions.jl ├── inference.jl ├── padding.jl ├── pooling.jl ├── runtests.jl ├── sampling.jl ├── softmax.jl ├── test_utils.jl ├── testsuite/ │ ├── fold.jl │ ├── gather.jl │ ├── rotation.jl │ ├── scatter.jl │ ├── spectral.jl │ └── upsample.jl └── utils.jl