Machine Learning Engineer

Bellatrix Aerospace is a pioneering space transportation company building next-generation propulsion systems and in-orbit solutions. Backed by the Government of India and leading investors, and recognized with two National Awards, we are shaping the future of space technology.

At Bellatrix, we foster a culture driven by innovation, curiosity, and resilience where failures are stepping stones and every challenge is an opportunity to create something truly ground-breaking.

We are looking for a Machine Learning Engineer with a strong focus on Computer Vision and Generative AI to drive innovation in ML optimization and edge deployment. In this role, you will transform cutting-edge research into real-world, high-impact applications in the space domain.

Design and develop state-of-the-art ML models for computer vision applications

Work on generative AI models (GANs, VAEs) and synthetic data generation

Implement computer vision techniques like object detection, segmentation, and tracking

Optimize models using quantization, pruning, and knowledge distillation

Deploy and scale models on edge devices (GPUs, TPUs, accelerators)

Build robust pipelines for model deployment and OTA updates

Benchmark, profile, and debug performance on target hardware

Convert models across platforms (TensorFlow Lite, ONNX, OpenVINO)

Simulate real-world deployment conditions (power, environment, connectivity)

Collaborate across teams and follow best practices in software development

Strong foundation in Mathematics (statistics, linear algebra, calculus) .

Proficiency in Python and libraries like NumPy, Pandas, scikit-learn .

Experience with TensorFlow, PyTorch, ONNX .

Expertise in computer vision and generative AI techniques.

Knowledge of model optimization for edge devices.

Familiarity with Git and software engineering practices.

Cloud platforms: AWS, GCP, Azure

Big data tools: Spark, Hadoop

MLOps: CI/CD, deployment, monitoring

&

Bachelors / Masters / PhD in Computer Science, Electronics, or related field

Proven 3 5 years experience in computer vision and generative AI projects

Strong understanding of deep learning architectures

Hands-on experience with large-scale datasets and preprocessing

Exposure to edge hardware (ARM, FPGAs, microcontrollers)

Excellent problem-solving, communication, and collaboration skills

?

Work on cutting-edge space technology with global impact

Be part of an award-winning, fast-growing startup

Turn research into real-world innovations

Thrive in a culture that values creativity, ownership, and growth

Collaborate with a passionate team building the future of space

: .