Senior Machine Learning Data Engineer

Builds data pipelines for ML models, optimizing edge and cloud performance.

• •The Axion Data team is building the data engine to train perception models, including edge applications and cloud pipelines for data ingestion and management.
• •This role involves optimizing ML model performance on edge and cloud, and building MLOps tooling.
• •Key Responsibilities Construct optimized data pipelines to run ML models.
• •Evolve data engine architecture to scale high-fidelity labels and accelerate ML iteration cycles.
• •Integrate foundation models (LLMs, VLMs) to automate and enhance labeling and data discovery.
• •Leverage software-in-the-loop and hardware-in-the-loop testing.
• •Interact with DoD customers to understand use cases and triage needs.
• •Requirements 5+ years of relevant work experience.
• •Familiarity with modern ML infrastructure, data-centric AI approaches, and running large-scale jobs on GPUs.
• •Experience creating or working on microservices and/or databases for data-oriented software.