v1.1.0

The Well v1.1.0 (April 4, 2025)

Enhancements

Metrics (944975e)

Added the following metrics:
- Mean absolute error
- Pearson Correlation

Updated the expected path structure for Well-formatted non-Well data to match Well data (944975e)

Users can now pass paths pointing to non-Well data produced in the Well format to use in the existing pipeline.

Augmentations (944975e)

Added the following augmentation options in a tensor-law consistent manner:

• Rotations
• Resize

Refactoring to allow for easier modification (944975e)

Users can now more easily extend the Well dataset objects for their own purposes as the function of __get_item__ has now been split into a small number of sub-components. Extension now only requires users to replace the components needed in their workflow rather than the full object.

Plotting updates (944975e)

• Power spectrum now generated for last timestep rather than average
• Bug fix in 3D (sliced) video
• Update "padding" BC mode to handle nD padding masks

Download dataset statistics along data (#22)

Previously, the statistics were included in the Github repo but not the data itself. This update now automatically downloads the statistics yaml files if they do not already exist in the given path.

Hugging Face integration

• Upload additional datasets (#31)
• Upload model checkpoints (#34, #35)

Delta vs full training within (18fcd05)

Now users can decide if they want to train models to predict the time step such that $x_{t+1} = x_t + f(x_t)$ or the full reconstruction where $x_{t+1} = f(x_t)$. Paper benchmarks are trained with full reconstruction, but predicting the delta/timestep is more effective in many applications.

Normalization changes also within (18fcd05)

Dataset normalization has been extended to include new options for both full and delta prediction. These now use parameterizable normalization modules for easy extension.

RMS and delta statistics addition (944975e)

Additionally, we've added new normalization options beyond what was in the paper including normalization by the root-mean-square of the field and applying normalization based on delta statistics.

Fixes

Bug fixes (18fcd05)

• Fix AFNO and AViT for non-square data
• Fixed normalization bug in test dataset. Now normalization is applied correctly in test rollouts.

Data fixes

• Added Rayleigh-Benard Uniform and fixed x dimension in existing data - Previously, the x dimension in Rayleigh-Benard was listed as being uniformly spaced. This data was actually sampled on Chebychev nodes so this spacing was not accurate. This has been addressed in two ways: 1. x dimension in the existing dataset has been updated to accurately reflect the spacing. 2. We have added a new rayleigh_benard_uniform that has been resampled onto a uniform grid from the underlying spectral representation of the solver. This is to ensure both backward compatibility while also enabled analysis of the uniformly spaced data.
• Replaced outlier trajectory in acoustic_scattering_inclusions - acoustic_scattering_inclusions chunk 11, position 47 had a corrupted trajectory. This has been replaced with a newly simulated valid example.
• Fix shear_flow documentation (#13)