Principal Software Engineer, AI/ML

Job Description:

Our Products Team is growing, and were looking for a highly skilled Principal Software Engineer, AI/ML, to join our cutting-edge AI / ML platform team. In this role, you will play a key part in designing, developing, and deploying advanced AI/ML models focused on content generation, natural language understanding, and creative data synthesis. You will work alongside a team of data scientists, software engineers, and AI/ML researchers to build systems that push the boundaries of what generative AI can achieve.

Key Responsibilities:

Design and Build ML Models: Develop and implement advanced machine learning models (including deep learning architectures) for generative tasks, such as text generation, image synthesis, and other creative AI applications.

Optimize Generative AI Models: Enhance the performance of models like GPT, V AEs, GANs, and Transformer architectures for content generation, making them faster, more efficient, and scalable.

Data Preparation and Management: Preprocess large datasets, handle data augmentation, and create synthetic data to train generative models, ensuring high-quality inputs for model training.

Model Training and Fine-tuning: Train large-scale generative models and fine-tune pre-trained models (e.g., GPT, BERT, DALL-E) for specific use cases, using techniques like transfer learning, prompt engineering, and reinforcement learning.

Performance Evaluation: Evaluate models performance using various metrics (accuracy, perplexity, FID, BLEU, etc.), and iterate on the model design to achieve better outcomes.

Collaboration with Research and Engineering Teams: Collaborate with cross-functional teams, including SME, AI researchers, data scientists, and software developers, to integrate ML models into production systems.

Experimentation and Prototyping: Conduct research experiments and build prototypes to test new algorithms, architectures, and generative techniques, translating research breakthroughs into real-world applications.

Deployment and Scaling: Deploy generative models into production environments, ensuring scalability, reliability, and robustness of AI solutions in real-world applications.

Stay Up-to-Date with Trends: Continuously explore the latest trends and advancements in generative AI, machine learning, and deep learning to keep our systems at the cutting edge of innovation.

Qualifications:

• Education: Bachelor s, Master s, or Ph.D. in Computer Science, Machine Learning, AI, Data Science, or a related field.

• Experience: 10+ years in cloud software development and 4 to 5+ years specializing in AI and Machine Learning

• Technical Expertise:

• Generative AI models (e.g., GPT, VAEs, GANs, Transformer architectures)

• Deep learning frameworks: TensorFlow, PyTorch, JAX

• Programming:

• Python (NumPy, Pandas, Scikit-learn), plus Java, Go, C/C++, R

• NLP, image generation, and multimodal models

• Training and fine-tuning large-scale models (e.g., GPT, BERT, DALL-E)

• Cloud platforms (AWS, GCP, Azure) and ML Ops (Docker, Kubernetes)

• Data engineering and large-scale dataset handling

• SQL and NoSQL databases

• Development Practices:

• Strong coding standards

• Testing and CI/CD pipelines

• Soft Skills:

• Ability to mentor and lead system design

• Excellent problem-solving, collaboration, and communication

• Proactive in learning and adopting new AI technologies

Preferred Skills:

• Experience with Reinforcement Learning or Self-Supervised Learning in generative contexts.

• Familiarity with distributed training and high-performance computing (HPC) for scaling large models.

• Contributions to AI research communities or participation in AI challenges and open-source projects.

• Tools: Linux, git, Jupyter, IDE, ML frameworks: Tensorflow, Pytorch, Keras, Scikit-learn

• GenAI: prompt engineering, RAG pipeline, Vector/Graph DB, evaluation frameworks, model safety, and governance

• India - (Flexible hybrid work model - work from Bangalore office for 20 days in a quarter)