Dictionary indexing with Numpy/Jax
I’m writing an interpolation routine and have a dictionary which stores the function values at the fitting points. Ideally, the dictionary keys would be 2D Numpy arrays of the fitting point coordinates, np.array([x, y])
, but since Numpy arrays aren’t hashable these are converted to tuples for the keys.