Application Software Engineer III — Machine Learning Systems - Project Sunrise – Orbital Data Centers
About The Position
This role is part of Advanced Concepts and Enterprise Engineering (ACE), supporting Blue Origin’s mission of enabling millions of people to live and work in space for the benefit of Earth. The team fosters innovation, advances engineering workflows, develops shared solutions and standards, and promotes simplicity, lower costs, and manufacturable design. The position supports Project Sunrise, a constellation of orbital data centers intended to transform how compute infrastructure is deployed and used. As an Application Software Engineer focused on Machine Learning Systems, you will help ensure customer AI models run efficiently on Blue Origin’s orbital platform. You will specialize in inference optimization, adapting and compiling complex neural networks for high performance within the platform’s power, thermal, and compute constraints. This is a technical software engineering role at the intersection of machine learning and aerospace. Applicants must be a U.S. citizen or national, a U.S. permanent resident (current Green Card holder), or lawfully admitted into the U.S. as a refugee or granted asylum. Travel may be required up to 10% of the time. Relocation assistance is available. A standard Blue Origin background check is required for all positions; additional checks may apply to certain job profiles.
Requirements
- Bachelor’s degree in Computer Science, Electrical Engineering, or a related technical field.
- At least 3 years of professional software engineering experience.
- Strong programming skills in Python and/or C++.
- Demonstrated experience with at least one major deep learning framework, such as PyTorch, TensorFlow, or JAX.
- Solid understanding of deep learning concepts and modern neural network architectures.
- Must be a U.S. citizen or national, U.S. permanent resident (current Green Card holder), or lawfully admitted into the U.S. as a refugee or granted asylum.
Nice To Haves
- Master’s degree or PhD in a relevant field.
- Direct experience with model optimization, quantization, and/or compilation for hardware accelerators such as GPUs or TPUs.
- Familiarity with NVIDIA tools such as TensorRT and Triton Inference Server.
- Experience with performance profiling and low-level code analysis.
- Passion for maximizing hardware performance.
- Knowledge of computer architecture.
Responsibilities
- Analyze and profile deep learning models to identify performance bottlenecks on on-orbit compute hardware.
- Use model optimization and compilation frameworks such as ONNX, TensorRT, and TVM to accelerate inference.
- Develop model quantization, pruning, and other optimization techniques for power- and thermally-constrained environments.
- Create and maintain a software toolkit and pipeline for onboarding and optimizing customer AI models.
- Collaborate with customers to understand their models and help achieve performance targets on the platform.
- Work with compute platform and flight software teams to understand hardware capabilities and constraints.
- Stay current with research and industry best practices in efficient deep learning and model optimization.
Benefits
- Medical insurance
- Dental insurance
- Vision insurance
- Basic and supplemental life insurance
- Paid parental leave
- Short- and long-term disability
- 401(k) with company match of up to 5%
- Education Support Program
- Stock options for regular employees working at least 20 hours per week
- Paid time off of up to four weeks per year, based on weekly scheduled hours
- Up to 14 company-paid holidays
- Some employees may be eligible for bonuses, depending on role type and job level; eligibility and amounts are not guaranteed.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
