Machine Learning Software Engineer

Stack

Responsibilities

• ▹Design, develop, and optimize computer vision models for real-time applications on embedded systems
• ▹Implement model compression techniques (quantization, pruning, knowledge distillation) to improve performance on low-power hardware
• ▹Deploy machine learning models on embedded platforms (ARM, NVIDIA Jetson, Qualcomm, or custom ASICs)
• ▹Write clean, efficient, well-documented code in Python and C++, using ML frameworks like TensorFlow, PyTorch, and ONNX
• ▹Develop and fine-tune SLAM, object detection, tracking, and feature extraction models for high efficiency
• ▹Collaborate with cross-functional teams to integrate ML models into production systems
• ▹Benchmark and profile ML models to identify optimizations for embedded hardware
• ▹Research and apply cutting-edge ML techniques to improve real-time performance in resource-constrained environments

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 designing and deploying optimized deep learning models for real-world applications
• ▹Proficiency in TensorFlow, PyTorch, ONNX, TensorRT, and other ML frameworks
• ▹Strong experience with model quantization, pruning, knowledge distillation, and hardware acceleration
• ▹Solid programming skills in Python and C++, with strong software optimization understanding
• ▹Familiarity with embedded platforms (NVIDIA Jetson, Raspberry Pi, ARM Cortex, Qualcomm AI accelerators)
• ▹Experience with hardware-aware model optimization to maximize inference speed
• ▹Strong problem-solving skills for complex technical challenges, working independently

Nice to have

• ▹Experience with real-time SLAM, visual odometry, and multi-sensor fusion
• ▹Knowledge of low-level hardware acceleration (CUDA, OpenCL, or specialized ML accelerators)
• ▹Familiarity with robotics frameworks (ROS) for integrating ML models
• ▹Background in edge AI deployments and optimizing neural networks for mobile and IoT devices

Soft skills

What we offer

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

About the company

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