How to Run the OMol25/UMA Models

by Corin Wagen · May 30, 2025

Meta's recent release of the OMol25 dataset and associated models makes it possible to run high-accuracy simulations without needing massive computing resources or specialized software. Last week, we wrote about the dataset, the models, and the effect that this was having on our users' research, and we've only heard more success stories since then.

There's a lot of demand for these models, but getting started with scientific simulation can be intimidating. Over the past few weeks, we've received a lot of questions about how scientists can use these models for their own research. In this post, we provide step-by-step tutorials on how to run the OMol25/UMA models, both locally and through Rowan's cloud chemistry platform.

Running Locally

1. Get Access To UMA/OMol25 Through HuggingFace

The OMol25 page on Hugging Face.

The model weights are stored on HuggingFace, but you have to request access and agree to the FAIR chemistry license, after which your request will be reviewed. Here's the link to OMol25, and here's the link to UMA.

If you're granted access, you can then download the model weights.

2. Install fairchem-core and ase

pip install ase
pip install fairchem-core

To run these calculations, install fairchem-core and ase (the Atomic Simulation Environment) from the Python Package Index.

3. Initialize the NNP

from ase.io import read
from fairchem.core import FAIRChemCalculator
from fairchem.core.units.mlip_unit import load_predict_unit

# to use an eSEN model:
esen_predictor = load_predict_unit(
    path="esen_sm_conserving_all.pt",
    device="cpu",
)

esen_calculator = FAIRChemCalculator(esen_predictor)

# to use an UMA model:

uma_predictor = load_predict_unit(
    path="uma-s-1.pt", 
    device="cuda",
        inference_settings="turbo", # optional ("turbo" accelerates inference but locks predictor to first system size it sees)
)
uma_calculator = FAIRChemCalculator(
    uma_predictor,
    task_name="omol", # options: "omol", "omat", "odac", "oc20", "omc"
)

Once the packages are installed, import them in Python and initialize a FAIRChemCalculator object using the model weights. (You'll need to change the path to match wherever you downloaded the weights.)

4. Build an ase.Atoms Object

atoms = read('benzene.xyz')  # or create ASE Atoms however you want
atoms.info = {"charge": 0, "spin": 1}

Structures can be loaded by creating an ase.Atoms object, which supports a variety of common filetypes. Full details about the different ways to create an ase.Atoms objects can be found in the ASE documentation.

5. Compute The Energy

atoms.calc = esen_calculator
energy_eV = atoms.get_potential_energy()
print(energy_eV) # -6319.18481262752

Finally, the FAIRChemCalculator can be associated with the Atoms object and get_potential_energy() can be called. This function uses the provided calculator to compute the single-point energy of the molecule, which is returned in eV.

The FAIRChemCalculator acts like any other calculator in ASE, making it possible to use this for geometry optimizations, molecular dynamics, and more.

Running Through Rowan

To quickly use the Meta models for high-level workflows, calculations can also be run through Rowan. Creating an account on Rowan is completely free and can be done using any Google-managed email account; create an account here.

1. Choose Workflow

Selecting the Basic Calculation workflow.

Once you sign in to Rowan, you can select which workflow you want to run. Here, we'll run a geometry optimization followed by a frequency calculation, both of which are accessed through the "Basic Calculation" workflow.

2. Input Molecule

Inputting a sucrose molecule.

Molecules can be loaded into Rowan by name, by SMILES, by input file, or through our provided 2D and 3D editors. Here, we'll input sucrose by name.

3. Select Level of Theory

Selecting the right level of theory.

To run a calculation using the Meta models, select the model from the “Level of Theory” dropdown menu. For now, Rowan only supports Meta's small eSEN model with conservative forces; more models will be added in the future.

4. Run Calculation

The finished calculation results.

Once you click “Submit Calculation,” we'll allocate a cloud GPU and start running the calculation. Results will start streaming to your browser almost immediately, and the output structures can be viewed, analyzed, and downloaded for further use.

Here, the optimization took 323 steps to complete and finished in 84 seconds (including the frequency calculation). Thermochemistry, vibrational frequencies, optimization steps, and overall calculation statistics can be viewed by navigating around the output page. Any optimization step can be resubmitted for further calculations, including scans, transition-state searches, intrinsic reaction coordinate calculations, and property-prediction workflows.

Banner background image

What to Read Next

ExpBDE54: A Slim Experimental Benchmark for Exploring the Pareto Frontier of Bond-Dissociation-Enthalpy-Prediction Methods

ExpBDE54: A Slim Experimental Benchmark for Exploring the Pareto Frontier of Bond-Dissociation-Enthalpy-Prediction Methods

ExpBDE54 is a benchmark dataset of experimental homolytic bond-dissociation enthalpies (BDEs) for 54 small molecules, used for benchmarking DFT, semiempirical methods, and NNPs.
Jul 17, 2025 · Jonathon E. Vandezande, Corin C. Wagen
Benchmarking Protein–Ligand Interaction Energy

Benchmarking Protein–Ligand Interaction Energy

How new low-cost computational methods perform on the PLA15 benchmark.
Jul 11, 2025 · Ishaan Ganti
Efficient Black-Box Prediction of Hydrogen-Bond-Donor and Acceptor Strength

Efficient Black-Box Prediction of Hydrogen-Bond-Donor and Acceptor Strength

Here, we report a robust black-box workflow for predicting site-specific hydrogen-bond basicity and acidity in organic molecules with minimal computational cost.
Jul 1, 2025 · Corin C. Wagen
Tracking Boltz-2 Benchmarks

Tracking Boltz-2 Benchmarks

Tracking the community's response to the new Boltz-2 model, plus some notes about Chai-2.
Jul 1, 2025 · Corin Wagen
g-xTB, Credit Usage, & More

g-xTB, Credit Usage, & More

the new g-xTB model from Grimme and co-workers; an easy visual overview of credit usage; better credit handling for organizations; bulk PDB download; a new collapsible JSON viewer
Jun 27, 2025 · Jonathon Vandezande, Ari Wagen, Spencer Schneider, and Corin Wagen
Representing Local Protein Environments With Atomistic Foundation Models

Representing Local Protein Environments With Atomistic Foundation Models

A guest post about how to use NNP embeddings for other prediction tasks.
Jun 20, 2025 · Meital Bojan and Sanketh Vedula
Co-Folding Updates

Co-Folding Updates

Boltz-2 FAQ and launch event recap; new visuals for co-folding workflows; new submission options; PDB bugfixes; new credit-management tools
Jun 12, 2025 · Ari Wagen, Spencer Schneider, and Corin Wagen
The Boltz-2 FAQ

The Boltz-2 FAQ

Questions and answers about the Boltz-2 biomolecular foundation model.
Jun 9, 2025 · Corin Wagen and Ari Wagen
Cleaning the Tap Room

Cleaning the Tap Room

beer and bezos; terms-of-service and privacy-policy updates; more deployment options; compliance requirements and country restrictions; a blog post about transition states
Jun 6, 2025 · Ari Wagen and Corin Wagen
BREAKING: Boltz-2 Now Live On Rowan

BREAKING: Boltz-2 Now Live On Rowan

This morning, a team of researchers from MIT and Recursion released Boltz-2, an open-source protein–ligand co-folding model.
Jun 6, 2025 · Corin Wagen, Spencer Schneider, and Ari Wagen