jax.experimental.pallas.multiple_of#

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

  • x (jax_typing.Array)

  • values (Sequence[int] | int)

Return type:

jax_typing.Array