Iteration helper */

The number of elements in the array. */

Broadcast the input arrays against one another while ignoring the * axes specified in `ignore_axes`. Note, this API is internal only. * The `ignore_

* Returns a function which computes the gradient of the input function with * respect to a vector of input arrays. * * The function being differ

Update a slice from the source array */

Complex mul followed by conjugate

Applies ``fn`` to the leaves of the Python tree ``tree`` and returns a new collection with the results. If ``rest`` is provided, every item i

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

Store results from simdgroup_matrix results into device memory */

Apply the gradients to the parameters and return the updated parameters. Can be used to update a model via ``model.update(opt.apply_g

Compute the one-dimensional Fourier Transform. */

Softmax of an array. */

Flattens a Python tree to a list of key, value tuples. The keys are using the dot notation to define trees of arbitrary depth and complexity.