Figure layout is broken

### Description

In the PDF, this figure shows a grid of plots, but the HTML version overlays the different grid cells in various sizes.

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

_No response_

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

Figures

### Internal issue ID

03de4a2d-f91e-4525-8fb7-1a6335bb8121

### Paper URL

https://arxiv.org/html/2403.08216v1

### Browser

Safari/605.1.15

### Device Type

Desktop

Open the paper at https://arxiv.org/html/2403.08216v1 in Safari and compare the affected figure with its PDF rendering. Identify which grid cells overlap in the HTML output; done means the figure preserves the PDF's grid layout with correctly sized, non-overlapping cells.

Written by the indexing model from the issue text.