Conversation
- Add Enzyme.jl dependency - Implement compute_grads() using Enzyme.autodiff with runtime activity mode - Add train/test mode switching to handle BatchNorm mutation issues - Refactor mlogloss to use direct indexing instead of onehotbatch - Configure Enzyme strictAliasing in module __init__ Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
…m Reactant cache Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
…rnal onehotbatch calls. Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
…neath, and default learners to Zygote/GPU for broader compatibility. Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
…odels.jl into jdb/multi-ad
* add support for multi-AD backend through extension Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * revert default to cpu Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * REMOVE ZYGOTE COMPAT Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * Move Reactant selection from to , keep Enzyme as the AD backend underneath, and default learners to Zygote/GPU for broader compatibility. Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> --------- Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
Contributor
|
Looks like a Julia |
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
* temporal embedding Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * cleanup imports Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * docstrings Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * temporal only fix Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * docstrings Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * temporal only setting fix Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * refactor(embeddings): IdentityEmbedding default, Parallel temporal branch, kwarg constructors Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * add embedding docs and use testmode Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> * fix piecewise binning for constant features Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com> --------- Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Migration from v0.3 to v0.4 by moving from Flux/Zygote to Lux/Reactant