Aarki · 19 hours ago
Staff Machine Learning Engineer (Models)
San Francisco, CA
Full-time
Onsite
Lead/Staff
8+ years expAarki is an AI-driven company specializing in mobile advertising solutions designed to fuel revenue growth. They are seeking a Staff Machine Learning Engineer to build, deploy, and test machine learning models for real-time advertising use cases. The role involves optimizing models for performance and partnering with engineering to ensure reliable deployments.
AdvertisingApp MarketingArtificial Intelligence (AI)MobileMobile Advertising
Responsibilities
Own the design, development, and operation of end-to-end machine learning models, including model training, evaluation, and deployment
Partner with engineering to productize models and ensure reliable deployments
Optimize inference performance across the stack, including latency and calibration issues
Design, execute, and analyze A/B tests and clearly communicate results
Perform root-cause analyses on model behavior, identify causes of prediction changes, and implement preventative measures
Design and implement new ML features across our model training and inference pipelines
Optimize our model architectures for a combination of inference speed and accuracy
Qualification
Required
8+ years of hands-on experience building and operating production machine learning systems
Strong Python and Spark for large-scale processing (on-prem/YARN environments preferred)
Proficiency in at least one systems language (e.g., C++, Java, Rust, Go) and strong experience with Python for ML development
Experience with data preparation and model training on large-scale data sets
Professional experience with deep learning frameworks
Familiarity with model serialization for serving, e.g., ONNX
Experience working with on-prem deployments of open source tools including Spark, ClickHouse, and Redash is a plus
Preferred
Exposure to online inference systems, gRPC/REST model endpoints, or streaming features (Kafka/Flink)
Ad-tech familiarity: auction dynamics, pacing, fraud signals, creative personalization
Company
Aarki
Arki is an AI company that offers advertising solutions aimed at enhancing revenue growth for mobile app businesses.
51-200 employees
H1B Sponsorship
Aarki 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 (4)
2024 (8)
2023 (6)
Funding
Current Stage
Growth StageTotal Funding
$1.74MKey Investors
Walden Venture Capital
2021-06-02Acquired
2013-05-02Convertible Note
2012-03-08Series A· $1.74M
Leadership Team
Aman Sareen
Chief Executive Officer
Recent News
2026-01-22
2025-06-05
Company data provided by crunchbase