Pierre Quinton

Hello @merlijn-sebrechts Thank you for the answer. The links points to various websites, could be youtube, github or other, but again it happened around 5-6 times during the last year...

@NicolasHug Thanks for your answer, yes this is exactly what I am looking for. I'm not sure if you would like to add something similar to torch or if you...

This holds only when $c\leq 1$, I guess you don't consider so much the case $c>1$, could you explain a bit why ? If $c>1$, it is possible that $g_w=0$...

> Tests are failing. But also, I don't know why we're passing a float in, that seems like a problem The test failing seems to be this one: https://github.com/pytorch/pytorch/blob/4c397e6ec6674881ea51ba67df35334065f2f5b6/test/functorch/test_eager_transforms.py#L2079 I...

Apparently, I missed a test with another float. Also lint failed with a line too long (even though I ran it locally). fixing those.

@zou3519 Sorry about this, I forgot to change the type of errors in the test. It made me realize that `ValueError` was also wrong, we check the type of `argnums`...

I'm getting bullied a bit by this unit test. It says that : ``` "jacfwd: `argnums` should be int or Tuple[int, ...], got: " "jacfwd: `argnums` should be int or...

@zou3519 I think this should pass the tests and be read to merge.

I would like to add that if someone wants JD in Jax and has the skills, then they can join the [discord server](https://discord.gg/76KkRnb3nk) to start working on it with us....

I added this when we where having 2 engines on the same modules. This is because the forward was then run using vmap with the functional api. This is exactly...