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
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
Location:
- India (Flexible hybrid work model - work from Bangalore office for 20 days in a quarter)
