Skip to content

Experiment with using out_axes=-1 inside nmap's vmap#99

Draft
shoyer wants to merge 1 commit into
google-deepmind:mainfrom
shoyer:vmap-end
Draft

Experiment with using out_axes=-1 inside nmap's vmap#99
shoyer wants to merge 1 commit into
google-deepmind:mainfrom
shoyer:vmap-end

Experiment with using out_axes=-1 inside nmap's vmap

89e09c1
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs