torch2jax_with_vjp (deprecated)

Deprecated: use torch2jax(..., depth=2) instead.

torch2jax_with_vjp

torch2jax.gradients.torch2jax_with_vjp(*args, depth=2, **kw)

Deprecated: use torch2jax(..., depth=2) instead.

Source code in torch2jax/gradients.py

def torch2jax_with_vjp(*args, depth=2, **kw):
    """Deprecated: use ``torch2jax(..., depth=2)`` instead."""
    warn_once(_WARN_TORCH2JAX_WITH_VJP_DEPRECATED, torch2jax_with_vjp)
    return torch2jax(*args, depth=depth, **kw)