Clarifies usage of jit on Module.init #910
jheek
commented
on 2021-01-20
Clarifies usage of using jax.jit in Module.init()
ae74bb3a
marcvanzee
changed the title Clarifies usage of jit(init) in WMT example Clarifies usage of jit on Module.init 4 years ago
jheek
commented
on 2021-01-25
Updates comment on jit(init)
7795d866
jheek
approved these changes
on 2021-01-27
Assignees
No one assigned
Labels
cla: yes
pull ready
Login to write a write a comment.
Login via GitHub