Umais comments

Repositories
Issues
Comments

Results 2 comments of


                                            Umais

vmap using in_axes doesn't handle named arguments

Has there been any update to this? I can confirm the issue still exists, see below: Vmapping a function with a keyword argument fails when using in_axes. Minimal example: ```python...

OSError Too many open files: /tmp/tmphv67gzd0wandb-media

I have also hit this issue - lost 12 long-running experiments, brutal!