Software Engineer, AI/ML Platform

Job Description

Software Engineer, AI/ML Platform

Company Overview

Join a pioneering technology company at the forefront of innovation, dedicated to creating transformative products and services that impact millions globally. We are a dynamic, fast-paced organization that thrives on creativity, collaboration, and continuous learning. Our mission is to empower the next generation of intelligent systems, and we’re looking for passionate individuals to help us achieve it.

Job Summary

We are seeking a talented and experienced Software Engineer to join our Artificial Intelligence and Machine Learning (AI/ML) Platform team. In this pivotal role, you will be responsible for designing, developing, and maintaining the core infrastructure, tools, and services that enable our data scientists and ML engineers to build, train, deploy, and manage machine learning models at scale. You will contribute to the foundational components that power our intelligent products, ensuring robustness, scalability, and efficiency across the entire ML lifecycle.

Key Responsibilities

• Design, develop, and implement scalable and reliable components of our AI/ML platform, including data pipelines, feature stores, model training infrastructure, inference services, and monitoring tools.
• Collaborate closely with data scientists, machine learning engineers, and product managers to understand their needs and translate them into platform features and improvements.
• Build and maintain robust MLOps tools and processes to automate the ML lifecycle, from experimentation and model versioning to continuous integration/continuous deployment (CI/CD) and production monitoring.
• Optimize the performance, reliability, and cost-efficiency of our distributed ML systems and infrastructure.
• Evaluate and integrate new open-source and proprietary technologies to enhance our platform capabilities and stay ahead of industry trends.
• Participate in code reviews, technical design discussions, and contribute to the overall architectural vision of the AI/ML platform.
• Troubleshoot and resolve complex technical issues related to the platform and its integrations across various environments.
• Create comprehensive documentation for platform features, APIs, best practices, and operational procedures.

Required Qualifications

• Education: Bachelor’s or Master’s degree in Computer Science, Engineering, or a related quantitative field.
• Experience: 3+ years of professional experience in software development, with a strong focus on building scalable backend systems, distributed systems, or platform infrastructure.
• Programming Skills: Expert proficiency in at least one modern programming language (e.g., Python, Java, Go).
• Cloud Platforms: Hands-on experience with major cloud providers (e.g., AWS, GCP, Azure) and their relevant services (e.g., S3, EC2, Kubernetes, SageMaker, Vertex AI).
• Distributed Systems: Solid understanding of distributed systems principles, microservices architecture, and API design.
• Containerization & Orchestration: Proficient with Docker and Kubernetes for containerizing and orchestrating applications.
• Databases: Experience with both SQL and NoSQL databases (e.g., PostgreSQL, MySQL, Cassandra, MongoDB).
• Problem Solving: Strong analytical, problem-solving, and debugging skills with a keen attention to detail.

Preferred Qualifications

• Experience building and scaling ML platforms or MLOps tools in a production environment.
• Familiarity with machine learning concepts, algorithms, and frameworks (e.g., TensorFlow, PyTorch, scikit-learn).
• Knowledge of data orchestration tools (e.g., Apache Airflow, Kubeflow Pipelines, Prefect).
• Experience with big data technologies (e.g., Apache Spark, Kafka, Flink).
• Familiarity with infrastructure-as-code tools (e.g., Terraform, CloudFormation).
• Contributions to open-source projects related to AI/ML or MLOps.
• Prior experience working in an agile development methodology.

Why Join Us?

• Impactful Work: Contribute to cutting-edge AI/ML projects that directly influence our product intelligence and user experience.
• Innovation Hub: Work with the latest technologies and a team passionate about pushing the boundaries of what’s possible in AI.
• Growth Opportunities: Benefit from a culture of continuous learning, professional development, and clear paths for career advancement.
• Collaborative Environment: Thrive in an inclusive and supportive team where your ideas are valued and collaboration is key.
• Comprehensive Benefits: Enjoy a competitive salary, generous stock options, comprehensive health, dental, and vision insurance, 401(k) matching, unlimited paid time off, and more.

To Apply

If you are a driven Software Engineer with a passion for building robust and scalable AI/ML platforms, we encourage you to apply! Please submit your resume and a cover letter detailing your relevant experience and why you are interested in this role through our careers portal. We look forward to hearing from you!

Equal Opportunity Employer

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. We are committed to providing an inclusive and accessible workplace for all employees.