slushbox
slushbox
Following a [comment] by @matrix4767 on the discussion post for [img2img-color-sketch], I implemented rudimentary functionality to let end users color on top of their uploaded images in the browser to...
I have this code: ```py shards = jax.sharding.PositionalSharding(np.array(jax.devices())).reshape(-1, 2) class ShardGetter: def __init__(self): self.transpose = True self.placement_cache = defaultdict(dict) def __call__( self, next_getter: callable, value: jax.Array, context: hk.GetterContext, ): if...
This request concerns a new pipeable operator: `prefetch`. Initially, it was just a utility I put together for a neural network data loader. However, it may also make a good...