[deleted] When we use @dr.wrap_ad decorator, how to use this with a method that has a drjit/torch input and other inputs of python primitives?

[Mephisto405]

When we use @dr.wrap_ad decorator, how to use this with a method that has a drjit/torch input and other inputs of python primitives?

For example, such method would be:

@dr.warp_ad(...)
def method(drjit_tensor, n_samples=10, some_boolean=False):
    pass

[Mephisto405]

Sorry, this works natively.