SqrtΒΆ In-place square root. Abstract Signature: Sqrt_(input: Tensor) PyTorchJAX (Core) PyTorchAPI: torch.sqrt_Strategy: Direct MappingOfficial Docs βJAX (Core)API: βStrategy: Macro 'jax.numpy.sqrt({input})'