Skip to content

OptimisersReactantExt: add support for RAdam#225

Open
csvance wants to merge 1 commit into
FluxML:masterfrom
csvance:reactant-radam
Open

OptimisersReactantExt: add support for RAdam#225
csvance wants to merge 1 commit into
FluxML:masterfrom
csvance:reactant-radam

Conversation

@csvance

@csvance csvance commented May 21, 2026

Copy link
Copy Markdown

This adds support for RAdam in Reactant.jl @jit / @compile by making the conditional use ifelse. This specialization should only be dispatched to when the input is traced.

Not sure documentation is needed here. From a user perspective RAdam should just work like any other optimiser when using Reactant.

PR Checklist

  • Tests are added
  • Documentation, if applicable

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant