3 ms·
Why is that? Why doesn't Jax just do something like class JaxWrapper: def __init__(self, arr): self.arr = arr def __setitem__(
by Evidlo 3y ago
Why is that? Why doesn't Jax just do something like
class JaxWrapper:
def __init__(self, arr):
self.arr = arr
def __setitem__(self, key, val):
return self.arr.at[key].set(val)
....