Senior Software Engineer – AI/ML Platform

Job Description

Senior Software Engineer – AI/ML Platform

At InnovateX Tech, we’re at the forefront of leveraging artificial intelligence to solve complex real-world problems. Our mission is to build intelligent systems that transform industries and improve lives. We foster a culture of innovation, collaboration, and continuous learning, empowering our engineers to push the boundaries of what’s possible. Join a dynamic team dedicated to creating cutting-edge AI solutions.

The Role

We are seeking a highly skilled and experienced Senior Software Engineer to join our Artificial Intelligence/Machine Learning (AI/ML) Platform team. In this pivotal role, you will be responsible for designing, developing, and maintaining scalable, robust, and high-performance infrastructure and tools that empower our data scientists and ML engineers to build, train, and deploy machine learning models efficiently. You will play a crucial part in shaping the future of our AI capabilities, working with the latest technologies in cloud computing, MLOps, and distributed systems.

Key Responsibilities

• Design, build, and optimize core components of our AI/ML platform, including data pipelines, model training frameworks, serving infrastructure, and monitoring systems.
• Develop and maintain tools and services that streamline the entire ML lifecycle from experimentation to production deployment (MLOps).
• Collaborate closely with data scientists, ML engineers, and product managers to understand their needs and translate them into platform features.
• Ensure the scalability, reliability, security, and cost-efficiency of the ML platform running on cloud infrastructure (e.g., AWS, GCP, Azure).
• Implement best practices for software engineering, including code reviews, automated testing, continuous integration/delivery (CI/CD), and documentation.
• Mentor junior engineers and contribute to the overall technical direction and architecture of the platform.
• Stay abreast of the latest advancements in AI/ML technologies, cloud services, and distributed systems to continuously evolve our platform.

What You’ll Bring

• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related technical field.
• 5+ years of professional software development experience, with a significant focus on building large-scale distributed systems or platform services.
• Strong proficiency in at least one modern programming language such as Python, Go, Java, or Scala. Python is highly preferred for ML platform roles.
• Demonstrable experience with cloud platforms (AWS, GCP, or Azure) and their relevant services (e.g., S3, EC2, Lambda, Sagemaker, GCP AI Platform, GKE, Azure ML).
• Solid understanding of data structures, algorithms, and distributed systems design principles.
• Experience with MLOps tools and concepts (e.g., MLflow, Kubeflow, Airflow, Docker, Kubernetes).
• Familiarity with machine learning concepts, model training, and deployment workflows.
• Excellent problem-solving skills, critical thinking, and a proactive approach to engineering challenges.
• Strong communication and collaboration skills, with the ability to work effectively in cross-functional teams.

Bonus Points

• Experience with big data technologies (e.g., Spark, Flink, Kafka).
• Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch).
• Contributions to open-source projects related to ML platforms or MLOps.
• Experience with infrastructure-as-code tools (e.g., Terraform, CloudFormation).

Why Join InnovateX Tech?

• Impact: Work on challenging and impactful projects that directly contribute to our core product and mission.
• Growth: Opportunities for continuous learning, professional development, and career advancement within a rapidly growing tech company.
• Culture: A vibrant, collaborative, and inclusive work environment where your ideas are valued.
• Benefits: Competitive salary, comprehensive health benefits, generous PTO, 401k matching, and more.
• Innovation: Be part of a team that encourages experimentation and stays at the cutting edge of AI technology.

How to Apply

If you are passionate about building robust and scalable AI platforms and thrive in a fast-paced, innovative environment, we encourage you to apply! Please submit your resume and a cover letter detailing your relevant experience to our careers portal.