Skip to content

feat: more nested AD rules#1151

Merged
avik-pal merged 5 commits intomainfrom
ap/nested_ad
Jan 1, 2025
Merged

feat: more nested AD rules#1151
avik-pal merged 5 commits intomainfrom
ap/nested_ad

Conversation

@avik-pal
Copy link
Member

@avik-pal avik-pal commented Dec 31, 2024 •
edited
Loading

fixes #1150

  • logsoftmax & softmax
    • Impl
    • Tests
  • MaxPool & MeanPool
    • Impl
    • Tests

@avik-pal avik-pal linked an issue Dec 31, 2024 that may be closed by this pull request
Efficient Jacobian Vector Products using ForwardDiff.jl #1007
Open
15 tasks
@github-actions
Copy link
Contributor

github-actions bot commented Dec 31, 2024 •
edited
Loading

Benchmark Results (ASV)

main 4dc08d1... main/4dc08d1eed50bf...
basics/overhead 0.129 ± 0.0015 μs 0.132 ± 0.0012 μs 0.979
time_to_load 0.89 ± 0.0094 s 0.89 ± 0.0068 s 1

Benchmark Plots

A plot of the benchmark results have been uploaded as an artifact to the workflow run for this PR.
Go to "Actions"->"Benchmark a pull request"->[the most recent run]->"Artifacts" (at the bottom).

@avik-pal avik-pal removed a link to an issue Dec 31, 2024
Efficient Jacobian Vector Products using ForwardDiff.jl #1007
Open
15 tasks
@avik-pal avik-pal merged commit 63d3434 into main Jan 1, 2025
32 of 58 checks passed
@avik-pal avik-pal deleted the ap/nested_ad branch January 1, 2025 00:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

CUDA 2nd order AD with MaxPool and logsoftmax

1 participant

@avik-pal