Senior AI Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

AIM · 10 hours ago

Senior AI Engineer

AIM is a well-funded, mission-oriented startup focused on building planetary-scale infrastructure and reversing climate change effects. They are seeking a Senior AI Engineer to advance their Machine Learning Platform by bridging the gap between Machine Learning and Software and ensuring a healthy lifecycle of ML models.

Artificial Intelligence (AI)ConstructionIndustrial AutomationInfrastructureMachine LearningMining TechnologyRobotics
check
H1B Sponsor Likelynote
Hiring Manager
Mike Martinsen
linkedin

Responsibilities

Design, build, and maintain automated CI/CD/CT pipelines for Machine Learning models. Design the deployment workflow to validate model safety before physical deployment
Optimize Machine learning models by building inference engines. You will manage GPU resource allocation on embedded devices to balance latency with power constraints
Provision and manage scalable cloud infrastructure to support distributed training clusters and simulation environments
Design and maintain a multi-modal data pipeline capable of ingesting and synchronizing massive streams of LiDAR, Camera, and IMU data. You will build the storage and retrieval layers that fuel our training sets
Implement robust monitoring to track model performance and data drift. You will be responsible for building telemetry to monitor system health and inference latency on the edge to ensure robot safety
Implement Feature Stores and strict Data Versioning to ensure total reproducibility of models. You will ensure we can trace every robot decision back to the exact code, data, and model version used
Maintain and evolve AIM’s ML data platform, including large-scale multi-modal data pipelines (LiDAR/Camera/IMU), schema/metadata standards, dataset indexing, and high-throughput storage + retrieval layers to support training, evaluation, and simulation workflows

Qualification

MLOpsPythonDeep Learning optimizationDockerKubernetesCloud architectureCI/CD toolsData pipelinesCUDAONNXTensorRTRobotics data formatsApache KafkaAirflowSparkROS / ROS2Embedded platformsML pipelines with simulators

Required

3+ years of professional experience in MLOps, DevOps, or Software Engineering, with a focus on productionizing ML models
Proficiency in Python and C++
Proven experience optimizing Deep Learning models with CUDA ONNX and TensorRT
Experience building large-scale ML data lakes and dataset pipelines (e.g., Ray Datasets / Ray Data) for ingestion, transformation, versioning, and retrieval of multi-modal training data
Strong command of Docker and Kubernetes
Hands-on experience with cloud architecture
Mastery of CI/CD tools
Familiarity with handling non-structured robotics data formats (e.g. Rosbags, MCAP, PCD files)
Experience with high-throughput data tools like Apache Kafka, Airflow, or Spark

Preferred

Experience working with ROS / ROS2
Experience deploying to embedded compute platforms
Experience integrating ML pipelines with simulators

Benefits

Company funded medical
Dental
Vision
401k
Life insurance
Gourmet food & perks

Company

AIM

twittertwittertwitter
company-logo
Scalable earthmoving is not optional. It’s required for our civilization to flourish.

H1B Sponsorship

AIM has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
2024 (3)
2023 (1)

Funding

Current Stage
Early Stage
Total Funding
$91.07M
2025-06-10Series A· $50M
2025-01-15Series Unknown· $41.07M
2021-10-03Undisclosed

Leadership Team

leader-logo
Adam Sadilek
CEO and Founder
linkedin
leader-logo
Robert Kotlaba
Co-Founder / Chief Hardware Engineer
linkedin
Company data provided by crunchbase