FluxML / FluxML/NNlib.jl

use vmap for all activations

Open
#220 0 comments 0 reactions 0 assignees View on GitHub
performance
Dominant language
Julia
Stars
258
Forks
138
Avg merge
2d 1h
Merged PRs (30d)
3

Description

i.e. extend #199 to all activations, now that Zygote implements the vmap adjoint.
Let's do proper benchmarking when doing this.

@AStupidBear

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reviewing issue #199 and the current activation implementations in NNlib.jl, then check how Zygote's vmap adjoint applies to each activation. Run the existing benchmarks before changing anything and compare results afterward; done means all activations use vmap with benchmark evidence recorded.

Written by the indexing model from the issue text.

Assessment

Tech stack
julia
Domain
machine-learning
Issue type
Refactor
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.