Vladimir Feinberg

Results 11 issues of Vladimir Feinberg

### Description ``` from functools import partial import jax from jax.experimental import pallas as pl import jax.numpy as jnp def add_vectors_kernel(x_ref, y_ref, o_ref): x, y = x_ref[...], y_ref[...] t =...

bug
pallas