jax.experimental.pallas.multiple_of

Contents

jax.experimental.pallas.multiple_of#

jax.experimental.pallas.multiple_of(x, values)[source]#

Parameters:

x (jax.Array)
values (Sequence[int] | int)

Return type: