Machine Learning Software Engineer
Swan, San Francisco
The role
We are seeking an expert Machine Learning Engineer with deep experience in computer vision, model optimization, and deployment on low-cost embedded systems. The ideal candidate will have a strong background in designing, training, and optimizing deep learning models for real-time applications. This role requires expertise in efficient neural network architectures, quantization, model compression, and hardware acceleration techniques to run ML models on resource-constrained devices.
What you will do
- 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.