Easy Apply
Easy Apply
As a Staff AI Engineer, you'll design and implement AI systems, develop APIs, and mentor engineers, collaborating with teams for impactful automation in advertising.
WHO WE ARE
At Basis Technologies, we empower agencies and brands with cutting-edge software that automates digital media operations. Our all-in-one platform supports seamless planning, reporting, and financial reconciliation across direct, programmatic, search, and social media.
We’re not just building advanced technology; we’re also building a culture where passionate, motivated individuals come together to drive meaningful change. Our commitment to inclusivity and growth ensures our employees thrive both professionally and personally.
With our headquarters in Chicago, we offer flexible work options across the U.S. and Canada, including remote, hybrid, or on-site at headquarters.
ABOUT THE TEAM
Technology is at the core of what we do. The AI Engineering team at Basis Technologies builds and scales machine learning and generative AI systems that power our programmatic advertising platform. We work closely with Data Science, Platform, and Application Engineering teams to deliver intelligent automation, real-time optimization, and deep performance insights across the media buying lifecycle.
Our mission is to deliver scalable, production-grade learning and inference systems that drive measurable impact and redefine what’s possible in programmatic advertising.
WAYS YOU’LL CONTRIBUTE
As a Staff AI Engineer, you’ll serve as a technical leader, designing and implementing the next generation of AI-powered systems across our platform. You will be both hands-on and strategic, balancing deep technical execution with architectural leadership. To be successful, you will contribute by:
- Collaborating with Data Science and Product teams to translate models into scalable, production-ready services.
- Integrating machine learning models and generative AI features throughout our media buying platform.
- Developing high-performance APIs, data pipelines, and infrastructure to support model training workflows and real-time inference over billions of daily events.
- Leading cross-team technical initiatives and establishing best practices for reliability, observability, and responsible AI.
- Influencing architectural decisions that shape the long-term direction of AI at Basis Technologies.
- Mentoring engineers and fostering a culture of code quality, technical excellence, and continuous learning.
WHAT YOU BRING TO THE TABLE
- 7+ years of experience in software engineering, with significant time spent delivering machine learning or generative AI systems in production.
- Proficiency in Python and Java, or similar languages.
- Experience designing and scaling APIs and data pipelines in distributed or cloud-native environments.
- Experience with both batch and real-time data processing frameworks such as Kafka, Spark, or Airflow.
- Proven ability to lead complex technical initiatives involving multiple teams or systems.
- Strong communication skills and a collaborative, impact-focused mindset.
- Commitment to creating inclusive, respectful environments where all voices are valued and supported.
BONUS POINTS
- Bachelor’s or advanced degree in Computer Science, Engineering, or a related field
- Prior experience in ad tech, large-scale recommender systems, or real-time bidding platforms
- Familiarity with MLOps principles, including model versioning, feature stores, A/B testing, and CI/CD for ML
- Experience with responsible AI practices, including bias mitigation and model explainability
- Experience with agentic AI systems, including autonomous agents, planning architectures, or tool-use frameworks
OUR TECH STACK
- Languages: Python, Java
- ML Frameworks:PySpark ML, ONNX
- Data & Messaging: Snowflake, Kafka, Airflow, SQS
- Infrastructure: AWS, EMR
ANYTHING ELSE?
Don’t have every skill listed? No problem! We know experience can be built in many ways. If you have relevant skills that aren't reflected in your resume, we encourage you to share them in an optional cover letter.
LIFE WITH BASIS TECHNOLOGIES
We’re committed to our people’s growth and well-being because our success is tied to theirs. That’s why we’ve earned recognition as a top workplace, including: Ad Age, #1 Best Places to Work 2025, WorkLife's #1 Most Committed to Work Life Balance, and the Chicago Tribune's Top Workplaces in Chicago.
We offer competitive perks, including a flexible work week, 401k/RRSP matching, mental health support, paid sabbaticals, generous parental leave, flexible work options, and more.
Basis is proud to be an equal-opportunity employer. We celebrate all team members regardless of gender identity, sexual orientation, race or cultural background, religion, disability, age, and beyond.
If you need assistance with interview accessibility, please contact [email protected]. Your privacy is important to us, view our policy here.
Top Skills
Airflow
AWS
Emr
Java
Kafka
Onnx
Pyspark Ml
Python
Snowflake
Sqs
Similar Jobs at Basis Technologies
AdTech • Digital Media • Marketing Tech • Software • Automation
The Senior Data Scientist will develop and implement machine learning and AI solutions, enhance product features, mentor team members, and collaborate across departments for improved customer outcomes.
Top Skills:
AirflowAWSKafkaOnnxPyspark MlPythonSnowflakeSparkSQL
AdTech • Digital Media • Marketing Tech • Software • Automation
The Client Analyst will manage client analytics, align campaign KPIs with business goals, analyze performance data, and present insights for campaign optimization.
Top Skills:
ExcelMS OfficePowerPoint
AdTech • Digital Media • Marketing Tech • Software • Automation
The Business Intelligence Developer will develop and implement BI solutions, design ETL processes, maintain reports, and collaborate with teams to leverage data for insights.
Top Skills:
AirflowConfluenceDomoGitJenkinsJIRAMongoDBMySQLPentahoPower BIPythonSnowflakeSQLStashTableauTalend
What you need to know about the Pune Tech Scene
Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.
