Published August 10, 2026
| Version v2
Dataset
Open
Training Dataset and Trained Model Weights for "Accelerating Chemical Kinetics for Exoplanet Atmospheres using Neural Networks"
Authors/Creators
- 1. Jet Propulsion Laboratory, California Institute of Technology
- 2. University of California, Santa Cruz
- 3. California Institute of Technology
- 4. Georgia Institute of Technology
- 5. NVIDIA Corporation
- 6. University of California, Riverside
Description
Trained model weights and dataset-provenance metadata accompanying the paper "Accelerating Chemical Kinetics for Exoplanet Atmospheres using Neural Networks" (Malsky et al., submitted to The Astrophysical Journal). Includes the published PyTorch Lightning checkpoint (best.ckpt), the exported deployable model (physical_model_k1_cpu.pt2), training metrics/logs, and the JSON records describing how the training/validation/test dataset was constructed and normalized (normalization.json, shard_index.json, preprocess_report.json, preprocessing_summary.json, dataset_config_snapshot.json). The emulator code is available at https://github.com/imalsky/Chemulator and the Robertson benchmark code at https://github.com/imalsky/robertson-emulator.
Files
dataset_config_snapshot.json
Files
(185.8 MB)
| Name | Size | |
|---|---|---|
|
md5:0487b63ebc1c3d9a9d2e230b380b9018
|
139.0 MB | Download |
|
md5:538d4008a57acd409d480dbcd1861feb
|
3.0 kB | Preview Download |
|
md5:e59533b99bf8e0a3302f15a981d5b9b9
|
74.7 kB | Download |
|
md5:bcb0699f4e8c0f34fa8778ce5a3a673d
|
19.3 kB | Preview Download |
|
md5:686e7c4c1cdf9a4147c74d3c2604cca9
|
18.6 kB | Preview Download |
|
md5:524bba1d82fc9eded4c9e3fbc4f67207
|
46.6 MB | Download |
|
md5:7d5692844a504ac942d0edb71c43fcec
|
1.3 kB | Preview Download |
|
md5:aa2e9411f20e68e71fb68d98aec6b260
|
14.0 kB | Preview Download |
|
md5:12af62ad63fd47d16f58ff3a3f1673b8
|
13.8 kB | Preview Download |
|
md5:d42e3b380dea54d9f1f7ed77268a780e
|
118.8 kB | Preview Download |
|
md5:336ede6bb1ad96caa4ea6e3fa213f954
|
10.8 kB | Download |