Incorrect position and size of images in a figure

### Description

In Figure 3 of the html version of the [paper](https://arxiv.org/html/2405.04890), certain subfigures (images) are positioned incorrectly, not respecting the set layout. Also, the last one's size is drastically increased which looks really terrible.

### (Optional:) Please add any files, screenshots, or other information here.

_No response_

### (Required) What is this issue most closely related to? Select one.

Choose One

### Internal issue ID

dab6b24c-5b2b-4862-82ea-fe7376eaee58

### Paper URL

https://arxiv.org/html/2405.04890

### Browser

Firefox/130.0

### Device Type

Desktop

Open the linked arXiv HTML paper in Firefox 130 on desktop and inspect Figure 3, comparing the subfigures with the intended layout. Trace the rendering or styling entry point responsible for figure images; done means every Figure 3 subfigure respects the layout and the final image no longer expands drastically.

Written by the indexing model from the issue text.