Swan

Bring the world to you!

Machine Learning Software Engineer

Full TimeRemoteTeam 11-50Since 2019Company SiteLinkedIn

Location

California

Posted

8 days ago

Salary

Not specified

Postgraduate Degree5 yrs expEnglishPythonPy TorchTensorflow

Job Description

• 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

Job 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

Related Job Pages