Senior ML Engineer - GPU-Accelerated PyTorch & GNNs

United States Digital Space LLC is seeking a senior software engineer to build large-scale ML solutions including Graph Neural Networks, Tabular Foundation Models, and ensemble models with GPU-accelerated training and inference. You will accelerate PyTorch-based frameworks and integrate CUDA-X libraries to power high-performance workflows.

You will collaborate with product, engineering, and science teams to develop GPU-accelerated implementations, mentor engineers, and drive architectural