Machine Learning Engineer

About Flyby Robotics:

Flyby Robotics builds drones with massively powerful GPU processing capabilities. Our team is crafting American-made drones that serve as fully open platforms for applications in machine learning, data capture, and autonomous operations. We envision a future where drones computing trillions of operations per second can be deployed by hundreds of thousands across the Western world with transformative impact across sectors, including industry, public safety, and defense. We are backed by top Silicon Valley investors, including Neo, MaC Ventures, Ryan Hoover (Product Hunt founder), Naval Ravikant (AngelList founder), and Balaji Srinivasan, among others.

The Role:

You'll work with a team of engineers and drone pilots from Yale University, NASA JPL, NVIDIA, and U.S. Naval Air Command at our Culver City production facility. We’re looking for a talented Machine Learning Engineer with expertise in computer vision and high-performance vision models to design, build, and deploy robust models that run efficiently on edge devices. As a foundational member of the machine learning team, you will play a key role in developing and optimizing machine learning pipelines for real-time inference in resource-constrained environments.

Key Responsibilities:

• Develop and fine-tune machine learning models (e.g. object detection, tracking, segmentation) for drone applications.
• Implement and maintain training pipelines using modern deep learning frameworks (e.g. PyTorch, Tensorflow, JAX).
• Optimize models for deployment on edge hardware (e.g. Nvidia Jetson).
• Conduct rigorous testing, including simulation and field testing on Flyby Robotics’ drone platforms.
• Continuously evaluate model performance and adapt to new datasets or mission requirements.

Core Requirements:

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field.
• 3+ years experience in machine learning and deep learning, with a focus on computer vision.
• Hands-on experience deploying models to edge devices (e.g., TensorRT, ONNX, OpenVINO).
• Proficiency in Python and deep learning libraries (e.g., PyTorch, JAX, TensorFlow).
• Strong understanding of model compression, quantization, and hardware-aware optimization.
• Experience with camera calibration, real-time video processing, and ROS or similar robotics frameworks.
• Experience writing production level C++ or Python code.
• Solid software engineering skills: Git, Docker, CI/CD workflows.
• We're an in-person hardware company!  
• Eligible to obtain an active U.S. security clearance.

Nice-to-Have:

• Experience working with drones or autonomous vehicles.
• Familiarity with 3D vision (stereo, depth estimation, SLAM).
• Knowledge of embedded systems and low-level optimization.
• Experience in simulation environments (e.g., Gazebo, AirSim).

Compensation and Benefits:

• $140,000 - $150,000 per year annual salary.
• 0.3-0.5% equity via stock options depending on the candidate's experience.
• Company holiday calendar, and competitive PTO plans.

You'll be building something incredibly consequential. If you like to push to see what you can accomplish with your grit and ingenuity, consider applying for this role at Flyby Robotics. 

Flyby Robotics is an equal-opportunity employer. We do not discriminate based on race, color, religion, national origin, gender, gender identity, gender expression, age, ancestry, citizenship, sexual orientation, status as a protected veteran, or any other legally protected characteristic.