
Senior AI Engineer with Databricks
Primary stack
Nice to have's
Job description
About the Position
We are seeking a Senior AI Engineer with Databricks to design, deploy, and operate production-grade machine learning systems, with a strong focus on feature engineering, model training, and real-time inference pipelines.
Responsibilities
- Design and optimize feature engineering pipelines for both batch and real-time workloads
- Build, deploy, and maintain scalable model training and inference pipelines
- Process structured and unstructured data at scale
- Develop streaming solutions using Spark Streaming
- Enable real-time feature generation and model serving
- Ensure reliability, scalability, and performance of ML solutions in production
- Support batch pipelines for model training and retraining
- Operate and maintain production ML systems with a strong track record of delivery
Requirements
- 5+ years of experience building, deploying, and operating offline and online feature engineering pipelines
- Strong experience with model training and real-time inferencing pipelines
- Hands-on expertise in Databricks, Spark/PySpark, Python, and SQL
- Experience processing large-scale unstructured data
- Strong knowledge of Spark Streaming
- Experience with real-time feature generation and low-latency model inference
- Proven track record of running production-grade ML systems
- English proficiency at B2 level or higher
Nice to Have
- Experience with MLflow and MLOps practices
- Familiarity with AWS
- Background in Customer Analytics, Recommendation Systems, or Retail
We Offer
-
Continuous Upskilling, Learning & Development
- Diversity of tasks and projects
- Assessment center for objective review of competency level
- Personal development plan
- Mentoring programs and leadership development
- Certification and professional development support
- Access to learning platforms including more than 2,500 internal courses
- English courses taught by certified teachers
-
Corporate Benefits
- Extra leave days
- Referral bonuses
-
Compensation Package
- Competitive compensation paid in USD
- Regular salary and performance reviews
-
Medical & Healthcare
- Private health insurance
- Well-being events
-
Working Environment
- Recreation areas and kitchens
- Tea, coffee, and snacks
- Sports equipment and game consoles
- IT Equipment
- Microsoft’s Software Assurance Home Use Program (HUP)
© EPAM. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.
EPAM helps organizations innovate their business processes and rethink the way they manage their businesses so they can remain competitive in this new digital age.
More at EPAM
All 3139 roles
Solution Architect - HCL Commerce
EPAM · India

Lead Java Automation Test Engineer with AI
EPAM · India

Lead Software Developer - Java with AWS Cloud
EPAM · India

Solution Architect – Java Full Stack
EPAM · India