Senior Software Engineer – AI/ML

Job Description

Senior Software Engineer – AI/ML

Location: San Francisco, CA | Job Type: Full-time | Experience Level: Senior

About Cognito AI Labs

Cognito AI Labs is a pioneering technology company at the forefront of artificial intelligence innovation. We are dedicated to creating intelligent solutions that transform industries, from healthcare to finance, by harnessing the power of advanced AI and machine learning. Our mission is to solve complex real-world problems and empower businesses with cutting-edge tools and insights. We foster a culture of curiosity, collaboration, and continuous learning, where every team member is encouraged to push boundaries and contribute to groundbreaking advancements. Join a diverse team of brilliant minds committed to making a tangible impact on the future.

The Opportunity

We are seeking a highly skilled and experienced Senior Software Engineer specializing in AI/ML to join our dynamic product development team. In this pivotal role, you will be instrumental in designing, developing, and deploying robust, scalable, and high-performance AI/ML models and systems. You will lead the charge in transforming complex data insights into actionable intelligence, driving innovation across our product suite. This is an exciting opportunity to work on challenging problems, mentor junior engineers, and contribute significantly to our core AI platform that powers our next-generation applications.

Key Responsibilities

• Architect and Develop AI/ML Solutions: Lead the end-to-end design, implementation, and deployment of advanced machine learning models and algorithms (e.g., deep learning, NLP, computer vision) for various product features.
• Data Pipeline Management: Build and optimize scalable data pipelines for training, validation, and inference of ML models, ensuring data quality, availability, and efficiency.
• Model Optimization & Deployment: Evaluate, fine-tune, and optimize ML models for performance, efficiency, and resource utilization, and integrate them into production systems using MLOps best practices.
• Code Quality & Best Practices: Champion clean code, robust testing, and maintainable software architecture. Conduct thorough code reviews and contribute to the evolution of our engineering standards.
• Research & Innovation: Stay abreast of the latest advancements in AI/ML research and actively explore new technologies and methodologies to enhance our products and platform capabilities.
• Collaboration & Mentorship: Work closely with product managers, data scientists, and other engineering teams. Mentor junior engineers, share knowledge, and contribute to a culture of continuous learning and excellence.
• Performance Monitoring & Troubleshooting: Design and implement monitoring systems for deployed ML models, troubleshoot issues, and ensure the reliability and stability of AI services.

Qualifications

Required Skills & Experience:

• Education: Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, or a related quantitative field.
• Experience: 5+ years of professional experience in software engineering with a strong focus on developing and deploying AI/ML systems in production environments.
• Programming Proficiency: Expert-level proficiency in Python, with experience in relevant ML frameworks (e.g., TensorFlow, PyTorch, Scikit-learn).
• ML Fundamentals: Deep understanding of machine learning algorithms, statistical modeling, and data structures.
• Cloud Platforms: Hands-on experience with cloud platforms like AWS, GCP, or Azure, particularly with ML-related services.
• Data Engineering: Experience with data manipulation libraries (e.g., Pandas) and database technologies (SQL, NoSQL).
• Software Engineering Principles: Strong grasp of software design patterns, object-oriented programming, and scalable system architecture.
• Communication: Excellent verbal and written communication skills with the ability to articulate complex technical concepts to diverse audiences.

Preferred Skills & Experience:

• Ph.D. in a relevant field.
• Experience with MLOps tools and platforms (e.g., Kubeflow, MLflow, Sagemaker).
• Familiarity with containerization (Docker) and orchestration (Kubernetes).
• Experience with big data technologies (e.g., Spark, Kafka).
• Demonstrated ability to lead technical projects or mentor junior team members.
• Publications or active contributions to the open-source ML community.

What We Offer

• Competitive Compensation: Attractive salary package, annual performance bonuses, and stock options in a fast-growing company.
• Comprehensive Benefits: Generous health, dental, and vision insurance plans, 401(k) matching, unlimited paid time off, and paid parental leave.
• Professional Growth: Dedicated budget for professional development, access to online courses, conference attendance, and opportunities for internal mobility and leadership.
• Dynamic Culture: A vibrant, inclusive, and collaborative work environment where innovation is celebrated, and ideas are encouraged. Regular team events, hackathons, and a focus on work-life balance.
• Impact: The chance to work on challenging, high-impact projects that shape the future of AI and contribute directly to our product success.
• State-of-the-Art Facilities: Modern office space with amenities, or full remote flexibility depending on the role and candidate preference.

Diversity & Inclusion

Cognito AI Labs is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We prohibit discrimination and harassment of any kind based on race, color, religion, national origin, sex, sexual orientation, gender identity, age, disability, veteran status, or any other protected characteristic as outlined by federal, state, or local laws.

How to Apply

Ready to make a difference? If you are a talented and passionate Senior Software Engineer with a deep understanding of AI/ML, we encourage you to apply! Please submit your detailed resume and a cover letter explaining your relevant experience and what excites you about this role via our careers portal at https://careers.cognitoailabs.com/senior-software-engineer-aiml. We look forward to reviewing your application!