Application Software Engineer III — Machine Learning Systems - Project Sunrise – Orbital Data Centers
Blue Origin3 days ago
Base Salary
$151k - $231k/yr
Responsibilities
- Analyze and profile deep-learning models to identify performance bottlenecks on orbital compute hardware.
- Use ONNX, TensorRT, TVM, and related compilation frameworks to accelerate inference.
- Develop quantization, pruning, and other model optimizations for power- and thermally constrained environments.
- Create and maintain a toolkit and pipeline for onboarding and optimizing customer AI models.
- Collaborate with customers to understand their models and achieve performance targets.
- Work with compute-platform and flight-software teams to understand hardware capabilities and constraints.
- Stay current with research and industry practices in efficient deep learning and model optimization.
- Travel may be required up to 10% of the time.
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++.
- 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, current U.S. permanent resident, or lawfully admitted refugee or asylee.
- Preferred: master’s degree or PhD in a relevant field.
- Preferred: experience with model optimization, quantization, or compilation for hardware accelerators such as GPUs or TPUs.
- Preferred: familiarity with NVIDIA TensorRT and Triton Inference Server.
- Preferred: performance profiling, low-level code analysis, and computer-architecture knowledge.
Benefits
- Medical, dental, vision, basic and supplemental life insurance, paid parental leave, short- and long-term disability, 401(k) matching up to 5%, and an Education Support Program.
- Stock options for regular employees working at least 20 hours per week.
- Up to four weeks of paid time off per year based on scheduled hours and up to 14 company-paid holidays.
- Relocation assistance is available.
- Some employees may be eligible for bonuses and additional benefits based on role type and job level; amounts and eligibility are not guaranteed.
Categories
About Blue Origin
Blue Origin designs and manufactures rocket engines, reusable launch vehicles, and lunar transport systems for commercial and government customers. Its products include the suborbital New Shepard, the in-development New Glenn orbital rocket, BE-4 engines (used on ULA’s Vulcan), and the Blue Moon lander; revenue comes from launch services, engine supply, and space systems contracts, including a NASA Human Landing System award. Founded in 2000 by Jeff Bezos, the company is headquartered in Kent, Washington.