True indicates the arrays buffer is safe to reuse */

Equivalence check defaults to false unless overridden by the primitive */

Equivalence check **/

r"""Join multiple schedules to create a new schedule. Args: schedules (list(Callable)): A list of schedules. Schedule :math:`i+1`

The Jacobian-vector product. */

Computes the Kullback-Leibler divergence loss. Computes the following when ``reduction == 'none'``: .. code-block:: python mx.

Computes the L1 loss. Args: predictions (array): The predicted values. targets (array): The target values. reduction

r"""Make a linear scheduler. Args: init (float): Initial value. end (float): Final value. steps (int): Number of steps to

Padded IO for Bluestein's algorithm

Load from device memory into threadgroup memory - with bound checking */

Four Step FFT First Step

Load from device memory into threadgroup memory - without bound checking */

Load from device memory into threadgroup memory - without bound checking */

Load from device memory into threadgroup memory - without bound checking */

r""" Computes the log cosh loss between inputs and targets. Logcosh acts like L2 loss for small errors, ensuring stable gradients, and li

r"""Applies the Log Sigmoid function. Applies :math:`\log(\sigma(x)) = -\log(1 + e^{-x})` element wise.