Machine Learning Engineer

Job Title: Machine Learning Engineer

Location: Beavercreek, OH or Remote/Hybrid]

Company: Oculii Corp. - A Subsidiary of Ambarella, Corporation

About Oculii Corp

Oculii is developing cutting-edge high-resolution imaging radar technology that powers the future of autonomous vehicles, drones, and intelligent machines. As a part of Ambarella, Inc., we integrate real-time AI with radar systems to push the limits of what's possible in autonomous perception.

Position Summary

We're seeking a Machine Learning Engineer to develop and deploy deep learning models that operate on radar data in resource-constrained environments. You'll build high-performance, scalable machine learning systems-from data preprocessing and model training to optimization and deployment on edge hardware.

Responsibilities

Build and maintain data pipelines for training and inference workflows.

Develop, train, and validate deep learning models (e.g., CNNs, Transformers).

Design and refine training strategies to improve performance and robustness.

Deploy models to embedded systems using optimization techniques (e.g., quantization, pruning).

Analyze and resolve performance bottlenecks in runtime environments.

Read and implement state-of-the-art research from academic and open-source communities.

Collaborate with multidisciplinary teams across ML, hardware, and software engineering. Required Qualifications

Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field.

Proficiency in Python and ML libraries such as PyTorch, NumPy, TorchVision, Matplotlib.

Understanding of deep learning architectures (e.g., MLPs, CNNs, Transformers).

Experience with model training tools: loss functions, optimizers, data loaders.

Proficiency in C/C++ for performance-critical components.

Solid foundation in statistics and data analysis.

Ability to interpret and implement techniques from ML research papers. Preferred Qualifications

Experience with deploying models using ONNX, TensorRT, or similar frameworks.

Familiarity with parallel computing and hardware acceleration (e.g., CUDA, SIMD, NPUs).

Experience with quantization, pruning, and model compression.

Exposure to real-time embedded systems or radar applications. Why Join Oculii?

Work with a highly skilled team solving real-world ML problems.

Contribute to impactful technologies in autonomy and intelligent sensing.

Be part of an innovation-focused company backed by Ambarella.

Competitive salary, benefits, and growth opportunities.

Oculii - Independently Build. Collectively Driven.

Come learn the way. Stay to lead the way.