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.
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
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
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 StageTotal Funding
$91.07M2025-06-10Series A· $50M
2025-01-15Series Unknown· $41.07M
2021-10-03Undisclosed
Recent News
2025-11-18
London Evening Standard
2025-10-20
Company data provided by crunchbase