Principal Software Engineer, ML Frameworks

Overview

Calix provides the cloud, software platforms, systems and services required for communications service providers to simplify their businesses, excite their subscribers and grow their value.

Job Description:

Our Products Team is growing, and we're 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
  
  
  

• 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)