[Feat] Add support for index tricks in autograd numpy

autograd/numpy/__init__.py

@@ -1,2 +1,7 @@
 from . import fft, linalg, numpy_boxes, numpy_jvps, numpy_vjps, numpy_vspaces, random
 from .numpy_wrapper import *
+from numpy import (
+    diag_indices_from, diag_indices, fill_diagonal, ndindex, ndenumerate,
+    ix_, c_, r_, s_, ogrid, mgrid, unravel_index, ravel_multi_index,
+    index_exp
+)