Skip to content

introduce custom_batching.sequential_vmap#9275

Merged
copybara-service[bot] merged 1 commit intojax-ml:mainfrom froystig:auto-vmapJan 28, 2022

Commits

Commits on Jan 21, 2022