Senior Software Engineer, AI Platform

Job Description

Senior Software Engineer, AI Platform

Join our innovative team as a Senior Software Engineer, AI Platform, where you will play a crucial role in designing, developing, and maintaining the core infrastructure that powers our cutting-edge artificial intelligence and machine learning initiatives. This position offers a unique opportunity to work on highly scalable, distributed systems, contributing directly to the evolution of our AI products and services that impact millions of users globally.

Key Responsibilities

• Architect & Develop: Lead the design, development, and implementation of robust, scalable, and high-performance AI platform components and services, including data pipelines, model training infrastructure, inference engines, and API layers.
• System Optimization: Continuously identify and implement optimizations for existing AI platform infrastructure, focusing on performance, reliability, cost-efficiency, and developer experience. Ensure seamless and efficient operation for data scientists and ML engineers.
• Cross-functional Collaboration: Work closely with data scientists, machine learning engineers, product managers, and other engineering teams to understand their needs, translate complex requirements into technical specifications, and deliver innovative, impactful solutions.
• Technical Leadership & Mentorship: Champion best practices for software development, including rigorous code reviews, automated testing, continuous integration/delivery (CI/CD), and comprehensive documentation. Mentor junior engineers and contribute to a culture of technical excellence.
• Troubleshooting & Support: Provide expert technical guidance and support, debugging complex issues across the AI platform, ensuring high availability and resilience of critical services.
• Innovation & Research: Actively research, evaluate, and prototype new technologies, tools, and methodologies to continuously improve our AI platform capabilities, scalability, and development processes, keeping us at the forefront of AI innovation.

Qualifications

• Education: Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related technical field.
• Experience: 5+ years of professional experience in software development, with at least 2-3 years specifically focused on building and scaling AI/ML infrastructure, data platforms, or large-scale distributed systems.
• Programming Proficiency: Expert-level proficiency in at least one of the following programming languages: Python, Go, or Java.
• Cloud Platforms: Hands-on, in-depth experience with major cloud platforms (AWS, GCP, or Azure) and their AI/ML services (e.g., AWS Sagemaker, GCP Vertex AI, Azure ML) and core infrastructure components (e.g., EC2/GCE, S3/GCS, Kubernetes, serverless functions).
• Distributed Systems: Strong understanding and practical experience with distributed systems principles, microservices architecture, containerization technologies (Docker, Kubernetes), and orchestration.
• Data Technologies: Experience with big data technologies (e.g., Apache Spark, Kafka, Flink) and various database systems (relational SQL, NoSQL like Cassandra, MongoDB, or DynamoDB).
• ML Frameworks: Familiarity with popular machine learning frameworks (e.g., TensorFlow, PyTorch, Scikit-learn) and concepts.
• Problem Solving: Excellent analytical and problem-solving skills, with a proven ability to tackle complex technical challenges independently and collaboratively, proposing robust and scalable solutions.
• Communication: Strong verbal and written communication skills, capable of clearly articulating complex technical concepts, designs, and decisions to both technical and non-technical audiences.

Preferred Skills

• Extensive experience with MLOps practices and tools (e.g., MLflow, Kubeflow, Airflow, ZenML).
• Knowledge of data governance, security, and compliance best practices in AI and big data environments.
• Demonstrated contributions to open-source projects related to AI/ML or distributed systems.
• Experience designing and implementing real-time inference systems and low-latency APIs.
• Familiarity with GPU programming, optimization techniques, and specialized hardware for AI workloads.
• Experience with streaming data processing and real-time analytics platforms.

Benefits

• Competitive salary and attractive equity options, reflecting your impact and experience.
• Comprehensive health, dental, and vision insurance plans for you and your family.
• Flexible working hours and remote-friendly options to support work-life balance.
• Generous paid time off, including vacation, sick leave, and company holidays.
• Dedicated budget for professional development, including conferences, online courses, and certifications.
• Matching 401(k) plan to help you plan for your future.
• Wellness programs, employee assistance initiatives, and access to mental health resources.
• Dynamic and inclusive company culture with regular team-building events, social gatherings, and hackathons.
• Opportunity to work on challenging problems that make a real-world impact.

Application Process

Interested candidates are invited to submit their resume and a concise cover letter detailing their relevant experience, motivations, and why they are excited about this role. Our hiring process typically includes an initial HR screening, a take-home technical assessment or live coding exercise, and a series of interviews with team members, technical leads, and leadership, focusing on technical depth, problem-solving, and cultural fit.

Our Culture

We are a fast-paced, innovative company driven by a passion for leveraging cutting-edge AI to solve complex, real-world problems across various industries. We foster a collaborative, intellectually stimulating, and supportive environment where curiosity, continuous learning, diverse perspectives, and bold ideas are highly valued. We believe in empowering our engineers to take ownership, innovate freely, and grow professionally. Join us to be part of a team that is not just building software, but shaping the future of AI.

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 building an inclusive and equitable workplace where everyone feels a sense of belonging.