Senior HPC ML & Image Processing Engineer

Socket.dev; San Jose (CA); Hybrid; USD 232,000 - 261,000; Full time

Benefits offered by this job

Equity compensation

Medical insurance

Dental insurance

Vision insurance

401k plan

Short & long term disability

Life insurance

Paid vacation

Paid holidays

Unlimited sick time

Paid parental leave

Job summary

Muon Space is seeking a senior architect to design and deploy a cloud image processing and ML serving platform. You will drive low-latency, scalable architectures for image data products and ML workloads, guiding engineers and setting standards for production code paths.

The role demands deep C++, PyTorch in production, Python interfacing, and strong cross-functional communication. Hybrid work with on-site San Jose collaboration is required.

Qualifications

• Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, Applied Math, or a related technical field.
• 6+ years of professional software engineering experience, including technical leadership on distributed computation or data-intensive systems.
• Expert-level C++, with memory management, concurrency, and performance-oriented programming in production systems.
• Strong PyTorch experience and production experience running models in C++ runtimes in the cloud using LibTorch or equivalent.
• Demonstrated experience architecting distributed systems for low-latency image and ML, data analysis, or computer vision.
• Strong Python skills, including interaction with native optimized implementations.
• Strong written and verbal communication and ability to lead cross-functional technical discussions.
• Ability and willingness to obtain and maintain a U.S. security clearance; Active clearance is a plus.

Responsibilities

• Architect, develop, align and deploy the cloud image processing and ML serving stack for Muon's ground data products, including runtime integration, request batching, and serving multiple customer data feeds.
• Design distributed, low-latency image processing systems that scale in the cloud.
• Define the patterns and interfaces by which data pipelines call into performant native/accelerated code.
• Set up a production-grade image processing and ML serving path for the Image Data Product pipeline, and extend the architecture to support upcoming imaging and onboard-compute missions and future onboard accelerator opportunities.
• Deliver high-performance reference algorithm implementations and benchmark them.
• Mentor other engineers on modern C++, image and ML runtimes, and distributed-systems design.
• Own verification and validation, including unit, integration, and performance regression tests.
• Produce clear designs, trade studies, and interface documentation.

Skills

C++

PyTorch

Python

Distributed systems

Leadership

Security clearance

Education

Bachelor's or Master's degree in CS/CE/EE or related field

Tools

LibTorch

CUDA

Job description

Muon Space is seeking a senior architect to design and deploy a cloud image processing and ML serving platform. You will drive low-latency, scalable architectures for image data products and ML workloads, guiding engineers and setting standards for production code paths.

The role demands deep C++, PyTorch in production, Python interfacing, and strong cross-functional communication. Hybrid work with on-site San Jose collaboration is required.