as_strided does very wild things with the Tensor, so it’s not surprising that vmap doesn’t like it.
If you are not using as_strided directly, you could see if you can get a traceback that says which function (or backward(?)) causes the error.
Best regards
Thomas