r"""Applies the Gaussian Error Linear Units function. .. math:: \textrm{GELU}(x) = x * \Phi(x) where :math:`\Phi(x)` is the Gaussian

r"""An approximation to Gaussian Error Linear Unit. See :func:`gelu` for the exact computation. This function approximates ``gelu`` with a m

r"""A fast approximation to Gaussian Error Linear Unit. See :func:`gelu` for the exact computation. This function approximates ``gelu`` with

Batch ndim > 1 */

Get CPU device name (brand string)

Can be tuned with MLX_MAX_OPS_PER_BUFFER, MLX_MAX_MB_PER_BUFFER

r"""Applies the gated linear unit function. This function splits the ``axis`` dimension of the input into two halves (:math:`a` and :math:`b`

Check if the array has an attached primitive or is a leaf node. */

r"""An initializer that returns an identity matrix. Args: dtype (Dtype, optional): The data type of the array. Default: ``float

Compute the two-dimensional inverse Fourier Transform. */

Return true if we are in a gradient trace (vjp, jvp, etc). */