Return a pointer to a compiled function

Extract sub-graphs of the graph that can be compiled and replace them with a Compiled Primitive.

Return whether NVRTC supports compiling to |device|'s SASS code.

3D convolution with a filter */

Conv helpers

1D transposed convolution with a filter */

A helper function to find the location of the current binary on disk. The Metal library ("mlx_ext.mtllib"), should be in the same directory.

Check if two sets are disjoint

Dtype from array protocol type string

Array protocol typestring for Dtype

* Approximation to the error function. * Based on code from: * https://stackoverflow.com/questions/35148198/efficient-faithfully-rounded-implementat

Compute exponential base e minus 1. Maximum ulp error = 0.997458 i = rint(a/log(2)), f = a-i*log(2). Then expm1(a) = 2**i * (expm1(f)+1) - 1. C

Extracts (weight, scales, biases) from Q4_0 tensors. Data layout is: |16 bit scale|32 x 4bit weights|.

Extracts (weight, scales, biases) from Q4_1 tensors. Data layout is: |16 bit scale|16 bit bias|32 x 4bit weights|.

Extracts (weight, scales, biases) from Q8_0 tensors. Data layout is: |16 bit scale|32 x 8bit weights|.

Compute the two-dimensional Fourier Transform. */