Skip to content

Add space-ml-sim (PyTorch fault injection for orbital AI)#161

Open
yaitsmesj wants to merge 1 commit intobharathgs:masterfrom
yaitsmesj:add-space-ml-sim
Open

Add space-ml-sim (PyTorch fault injection for orbital AI)#161
yaitsmesj wants to merge 1 commit intobharathgs:masterfrom
yaitsmesj:add-space-ml-sim

Conversation

@yaitsmesj
Copy link
Copy Markdown

Adds space-ml-sim to the Other libraries section.

PyTorch-specific features:

  • Bit-flip fault injection into PyTorch model weights and activations, driven by radiation-derived Poisson rates
  • Transformer-aware targeting: separate handling for attention, LayerNorm, and embedding layers
  • Full TMR and selective TMR (per-layer vulnerability ranking) via TMRWrapper
  • Checkpoint rollback with majority voting and anomaly detection
  • ONNX import for fault injection on models not written in native PyTorch
  • Quantization comparison (FP32 / FP16 / INT8 fault resilience curves)
  • Per-layer sensitivity heatmap for vulnerability analysis

Broader context: Built around an orbital mechanics + radiation environment model (Walker-Delta constellations, SGP4 TLE propagation, SEU/TID rates, SAA, solar cycle). Lets PyTorch users stress-test model robustness under realistic space radiation — a use case not currently represented on this list.

Stats: 497 tests, 80%+ coverage, validated against published SEU measurements (ISS, sun-sync EO, high-LEO) and SPENVIS reference data. Open-source (AGPL-3.0), on PyPI.

space-ml-sim simulates AI inference on orbital satellite
constellations under space radiation. It provides PyTorch
bit-flip fault injection driven by radiation-derived Poisson
rates, transformer-aware targeting, Triple Modular Redundancy,
and checkpoint rollback for reliability studies.
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