Cos

In-place cosine operation.

Abstract Signature:

Cos_(input: Tensor)

PyTorch

API: torch.cos_
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.cos
Strategy: Plugin (unroll_inplace_ops)

NumPy

API: numpy.cos
Strategy: Plugin (unroll_inplace_ops)

Flax NNX

API: jax.numpy.cos
Strategy: Plugin (unroll_inplace_ops)

PaxML / Praxis

API: jax.numpy.cos
Strategy: Plugin (unroll_inplace_ops)