Swan

Machine Learning Software Engineer

Swan

full-time

Posted on:

Location Type: Remote

Location: California • United States

Visit company website

Explore more

✨ AI Apply
Apply

Tech Stack

PythonPyTorchTensorflow

About the role

  • 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 to improve performance on low-power hardware
  • 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++, leveraging 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, optimizing for latency, accuracy, and power consumption
  • 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 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 in 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 techniques
  • Solid programming skills in Python and C++, with a strong understanding of software optimization
  • Familiarity with embedded platforms such as NVIDIA Jetson, Raspberry Pi, ARM Cortex, Qualcomm AI accelerators, or specialized AI chips
  • Experience with hardware-aware model optimization to maximize inference speed and minimize memory footprint
  • Strong problem-solving skills and ability to work independently on complex technical challenges
Benefits
  • Top-tier health, dental, vision, short-/long-term disability, and life insurance, with full employee coverage and partial coverage for dependents
  • Flexible/reasonable vacation and sick leave
  • 401(k) plans, FSA, HSA, and commuter benefits
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
computer visionmodel compressionquantizationpruningknowledge distillationmachine learningdeep learningSLAMobject detectionfeature extraction
Soft Skills
problem-solvingcollaborationindependent work
Certifications
Master’s degreePh.D.