Skip to content

Add plot module, batch gRASPA setup, UMA batch, and zeopp#6

Merged
tdpham2 merged 5 commits into
mainfrom
feature/add-plot-module
May 26, 2026
Merged

Add plot module, batch gRASPA setup, UMA batch, and zeopp#6
tdpham2 merged 5 commits into
mainfrom
feature/add-plot-module

Conversation

@tdpham2

@tdpham2 tdpham2 commented May 12, 2026

Copy link
Copy Markdown
Owner

Summary

  • Add setup_batch() for gRASPA: set up simulations across a CIF folder x temperature x pressure grid, with simulations.jsonl manifest for Parsl dispatch
  • Add matkit graspa batch-setup CLI command with repeatable --temp and --pressure flags
  • Add multi-GPU batch optimization for UMA models with queue-based GPU worker distribution
  • Add Zeo++ module for pore geometry analysis with single-file and batch modes
  • Add plot module with isotherm, mixture, and selectivity plotting
  • Fix output parsing bugs across graspa, graspa_sycl, raspa2, and mlip modules
  • Add CI, tests, and improve overall code quality

Test plan

  • All 10 graspa tests pass
  • Smoke tested gRASPA batch setup across 2 temps x 2 pressures
  • Verify matkit graspa batch-setup --help shows all options
  • Test on a real CIF folder with multiple structures
  • Verify matkit mlip uma-opt-batch --help shows all options
  • Run full test suite: pytest tests/

tdpham2 added 3 commits May 11, 2026 22:17
Add run_opt_uma_batch() with queue-based multi-GPU execution using
multiprocessing spawn context. Workers cache calculators per
(model, task_name) to avoid reloading. Supports single file or
directory input with Cartesian product of models x run_types.
Includes separate fmax/fmax_cell parameters, results.jsonl output,
and uma-opt-batch CLI command.
Support setting up simulations for all CIF x temperature x pressure
combinations from a single call. Writes a simulations.jsonl manifest
for downstream Parsl dispatch.
@tdpham2 tdpham2 changed the title Add plot module, zeopp, UMA batch, and code quality improvements Add plot module, batch gRASPA setup, UMA batch, and zeopp May 26, 2026
@tdpham2 tdpham2 merged commit 68631b0 into main May 26, 2026
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant