Machine Learning Engineer for Large Scale AI Project (Computer Vision)

About OneSec

OneSec is a leading innovator in drone technology, specializing in AI powered autonomous delivery drones. We are dedicated to pushing the boundaries of what's possible in aerial robotics and control systems. Situated near the New Belgrade trainstration, committed to advancing the capabilities of aerial vehicles.
Check out our website OneSec.com for more info

Job Description

OneSec is building one of the largest machine learning projects in the drone industry. With our own photorealistic flight simulator, and a fleet of prototype drones, we are building one of the largest depth perception datasets in the world, with 150 million training samples.

The job has 2 main components:

Primary: Dataset generation: using open source AI models to generate pseudo labels for a large image dataset. Generating synthetic stereo images from images with depth map. Filtering large public datasets like LAION-5B. 
Secondary: Train models, modify model architecture etc.

Requirements

• Strong expertise in deep learning
• Experience with computer vision and CNNs (resnets etc)
• Experience with PyTorch & CUDA
• Good generalist tech skills: cloud infrastructure, image processing etc. 

Why Join OneSec?

• Work in an exceptionally talented team
• Attractive salary & stock option package
• Work on cutting-edge autonomous drone technology
• State-of-the-art development tools & equipment
• Office directly at New Belgrade trainstation (easy commuting from other towns)