pygrad.numba_ops

Module storing numba-defined optimizations.

Functions

combined_softmax_backprop(new_val, new_grad)
conv2d_bwd(kernel, signal, grad)
conv2d_fwd(kernel, signal)
find_sum_grad_3D(new_val)
mat_sum(mat1, mat2)
softmax_grad(new_val, grad)
softmax_matmul_3D(A, B)