AI Architect

Job Title: AI Architect

Location: [Pune, Onsite]

Experience: 6+ Years

Employment Type: Full-Time

Working requirements:

Engagement Commitment: 3 months to start with to ensure that the engagement is progressing well.

Deliverable: Minimum Viable Product is already there, enhancement is required

Working Hours Preferable: California Time (which is ~12 hours behind IST), requesting this because then it will be easier to huddle and both US and India team can work simultaneously

About the Role:

We are seeking an experienced and visionary AI Architect to lead the design, development, and deployment of cutting-edge AI/ML solutions. The ideal candidate will have a strong foundation in artificial intelligence, machine learning, data engineering, and cloud technologies, with the ability to architect scalable and high-performing systems.

Key Responsibilities:

• Design and implement end-to-end AI/ML architectures for large-scale enterprise applications.
• Lead AI strategy, frameworks, and roadmaps in collaboration with product and engineering teams.
• Guide the development of machine learning models (e.g., NLP, Computer Vision, Predictive Analytics).
• Define best practices for model training, validation, deployment, monitoring, and versioning.
• Collaborate with data engineers to build and maintain robust data pipelines.
• Choose the right AI tools, libraries, and platforms based on project needs (e.g., TensorFlow, PyTorch, Hugging Face).
• Work with cloud platforms (AWS, Azure, GCP) to deploy and manage AI models and services.
• Ensure AI/ML solutions comply with data privacy, governance, and ethical standards.
• Mentor junior AI engineers and data scientists.

Required Skills & Qualifications:

• Bachelor's or Master’s degree in Computer Science, Data Science, AI/ML, or a related field.
• 6+ years of experience in AI/ML, with at least 2+ years in an architecture or lead role.
• Strong experience with AI/ML frameworks: TensorFlow, PyTorch, Scikit-learn, etc.
• Deep understanding of LLMs, transformers, GPT models, and fine-tuning techniques.
• Proficiency in Python and data processing libraries (Pandas, NumPy, etc.).
• Experience with cloud-based AI services (AWS SageMaker, Azure ML, Vertex AI, etc.).
• Knowledge of MLOps practices, CI/CD for models, and model monitoring.
• Familiarity with data lakehouse architecture, real-time inference, and APIs.
• Strong communication and leadership skills.

Preferred Qualifications:

• Experience with generative AI applications and prompt engineering.
• Knowledge of reinforcement learning or federated learning.
• Publications or contributions to open-source AI projects.
• AI certifications from cloud providers (AWS, Azure, GCP).

Why Join Us?

• Work on transformative AI projects across industries.
• Collaborate with a passionate and innovative team.
• Flexible work environment with remote/hybrid options.
• Continuous learning, upskilling, and growth opportunities.