Normalize the str to lower case and make them float numbers if possible.

Reshape an image of [h, w, 3] -> [n_patches, pixels_per_patch]

Returns a list of FSDP modules with their FQN.

Handles some cases where the state dict is valid yet may need to be transformed in order to be loaded. This modifies the sta

Reshape images of [n_images, h, w, 3] -> [n_images, n_patches, pixels_per_patch]

Modify ``x`` in place to replace ``float("-inf")`` with the minimum value of the dtype when ``check_neg_inf`` is ``True`` and to replace ``fl

Gets the default evaluation config for task (or task:split string) `name`

Build a HTML element containing `image_src` and the boxes in `boxes` on top of it. Provides a way to draw annotated images without have to load/m

Max numbers of that can be built for one image

Sample `Gumbel(phi)` conditioned on the maximum value being equal to `T`. `phi` should have shape `(batch_size, num_classes)` and `T

Check if the point (x, y) is within the region defined by the boolean mask. Parameters: - point (tuple of floats): x/y-coordinate of the

Load model state from a distributed FSDP model checkpoint created from :func:`save_fsdp_model_and_optim_state()`. Note that ``model`` should

Seed all rng objects.