RandInt ======= Auto-generated from jax_code_defs .. raw:: html

PyTorch

API: torch.randint
Strategy: Direct Mapping

JAX (Core)

API: jax.random.randint
Strategy: Plugin (inject_prng)

NumPy

API: numpy.random.randint
Strategy: Direct Mapping

Keras

API: keras.random.randint
Strategy: Direct Mapping

TensorFlow

API: tf.random.uniform
Strategy: Direct Mapping

Apple MLX

API: mlx.random.randint
Strategy: Direct Mapping

Flax NNX

API: nnx.rnglib.Rngs.randint
Strategy: Direct Mapping