* Parse a sockaddr from an ip and port provided as strings. */

Do necessary transposes and copies to prepare the inputs and outputs for building the cuDNN conv op. It is safe to be called multiple times in one eva

Register inputs and outputs before actually running conv op. Can only be called once per eval_gpu.

Main kernel function */

Main kernel function */

Save the model's weights to a file. The saving method is determined by the file extension: - ``.npz`` will use :func:`mx.savez`

Create a cuDNN tensor for scalar.

r"""Applies the Softplus function. Applies :math:`\log(1 + \exp(x))` element wise.

Store results from simdgroup_matrix results into device memory */

Split string into array.

Create a 4D cuDNN tensor from 1D array, with |axis| being contiguous dim.

Applies ``fn`` to the path and leaves of the Python tree ``tree`` and returns a new collection with the results. This function is the same :f

Unfreeze the Module's parameters or some of them. This function is idempotent ie unfreezing a model that is not frozen is a noop.

Get original kv from slices, i.e. undo keys[..., :offset, :]

Not copyable or moveable