Skip to content

flax is way too slower in speed and not even converging, than pytorch counterpart it took 30min in flax, whereas in pytorch took 2min and converged way earlier #4717

Unanswered
ak24watch asked this question in Q&A
flax is way too slower in speed and not even converging, than pytorch counterpart it took 30min in flax, whereas in pytorch took 2min and converged way earlier #4717
@ak24watch ak24watch
Apr 19, 2025 · 0 comments
Return to top
Discussion options

ak24watch
Apr 19, 2025

You must be logged in to vote

Replies: 0 comments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant
@ak24watch