Machine Learning Software Engineer

About the Company

Swan delivers the future of defense and industry through scalable autonomous products. Swan is backed by leading defense VCs including a16z's American Dynamism fund.

Responsibilities

• Design, develop, and optimize computer vision models for real-time applications on embedded systems
• Implement model compression techniques such as quantization, pruning, and knowledge distillation
• Deploy machine learning models on embedded platforms including ARM, NVIDIA Jetson, Qualcomm, or custom ASICs
• Write clean, efficient, and well-documented code in Python and C++
• Develop and fine-tune SLAM, object detection, tracking, and feature extraction models
• Collaborate with cross-functional teams to integrate ML models into production systems
• Benchmark and profile ML models to identify and implement optimizations for inference on embedded hardware
• Research and apply cutting-edge ML techniques to improve real-time performance

Requirements

• Master’s or Ph.D. in Computer Science, Electrical Engineering, Machine Learning, or a related field
• 5+ years of experience in machine learning, deep learning, and computer vision
• Extensive experience in designing and deploying optimized deep learning models
• Proficiency in TensorFlow, PyTorch, ONNX, and TensorRT
• Strong experience with model quantization, pruning, and knowledge distillation
• Solid programming skills in Python and C++
• Familiarity with embedded platforms such as NVIDIA Jetson, Raspberry Pi, ARM Cortex, or Qualcomm AI accelerators
• Experience with hardware-aware model optimization

Preferred Qualifications

• Experience with real-time SLAM, visual odometry, and multi-sensor fusion
• Knowledge of low-level hardware acceleration using CUDA or OpenCL
• Familiarity with robotics frameworks such as ROS
• Background in edge AI deployments and optimizing neural networks for mobile and IoT devices

Benefits

• Top-tier health, dental, vision, short-/long-term disability, and life insurance
• Flexible/reasonable vacation and sick leave
• 401(k) plans, FSA, HSA, and commuter benefits