AI/ML Engineer

AI/ML Engineer - Fresher/Entry Level
Surviant | Full-Time | Hyderabad, India (On-site)

About Surviant

Surviant is a Digital Innovation Studio that specializes in building cutting-edge AI-powered products for developers and enterprises. We solve complex problems at the intersection of artificial intelligence and developer tooling, creating solutions that make developers more productive and codebases more maintainable.

Our team of AI engineers, backend developers, and product specialists works on next-generation code intelligence systems. We focus on graph-based architectures, machine learning pipelines, and real-time inference systems that push the boundaries of what developer tools can do. From neural network optimization to production-grade ML infrastructure, we're committed to building AI that actually ships and scales.

The Role

We're seeking a motivated AI/ML Engineer to join our core product team and help build the ML backbone of our developer tools platform. This is an excellent opportunity for a fresher or early-career engineer with strong fundamentals to grow their skills while working on genuinely challenging AI problems. You'll work directly with our founding team, contribute to production ML systems, and help shape the technical direction of our AI products.

Key Responsibilities

ML Pipeline Development

• Build and optimize machine learning pipelines using PyTorch
• Implement and fine-tune neural network architectures for code understanding
• Develop embedding models and similarity search systems
• Work on graph-based neural network architectures (GNNs)

Model Optimization & Deployment

• Optimize inference for real-time performance (sub-300ms targets)
• Implement efficient caching and batching strategies for ML models
• Support model deployment and integration with backend services
• Monitor model performance and iterate based on real-world usage

Research & Experimentation

• Read and implement ideas from recent ML research papers
• Run experiments to validate new approaches and architectures
• Collaborate with the team on technical design decisions
• Document findings and contribute to internal knowledge base

Cross-Functional Collaboration

• Work closely with backend engineers on ML integration
• Participate in code reviews, design discussions, and sprint planning
• Communicate technical concepts clearly to non-ML team members
• Contribute to a collaborative, learning-focused team culture

Required Qualifications

• Education:
  Bachelor's degree in Computer Science, AI/ML, or related field (or equivalent practical experience)
• Python Skills:
  Strong proficiency in Python programming
• PyTorch Experience:
  Hands-on experience with PyTorch (academic projects, personal projects, or internships count)
• ML Fundamentals:
  Solid understanding of neural networks, embeddings, loss functions, and training loops
• Mathematics:
  Comfortable with linear algebra, probability, and calculus fundamentals
• Self-Learning:
  Ability to read ML papers and implement ideas independently
• Problem-Solving:
  Strong analytical thinking and debugging skills
• Communication:
  Ability to explain technical concepts clearly

Preferred Qualifications

• Experience with Graph Neural Networks (DGL, PyTorch Geometric)
• Familiarity with code models (CodeBERT, GraphCodeBERT, CodeT5)
• Understanding of contrastive learning or representation learning
• Experience with AST parsing or static code analysis
• Exposure to model optimization techniques (quantization, pruning, distillation)
• Contributions to open-source ML projects
• Experience with ML experiment tracking (Weights & Biases, MLflow)
• Understanding of transformer architectures and attention mechanisms

What We Offer

Professional Growth

• Base salary starting at ₹3,00,000 per annum (negotiable based on skills, we assess fit during the interview process)
• Direct mentorship from founding team and senior engineers
• Hands-on experience building production ML systems from day one
• Learning budget for courses, conferences, and resources
• Clear growth path with opportunities to lead ML initiatives

Benefits & Compensation

• Performance-based bonuses tied to product milestones
• Equity/ESOP potential for high performers
• Flexible working hours
• Health insurance coverage

Work Environment

• On-site position at our Hyderabad office
• Small team with high ownership and direct impact
• Modern ML stack with access to GPU resources
• No bureaucracy — ship fast, learn faster
• Collaborative culture focused on building great products

Application Process

Ready to build AI that developers actually use? We'd love to hear from you

To Apply:

1. Submit your resume highlighting relevant projects and coursework
2. Include links to GitHub, Kaggle, or any ML projects you've worked on
3. Brief note on why you're excited about AI and developer tools

Selection Process:

• Round 1: Resume screening and initial call
• Round 2: Technical assessment (ML fundamentals + coding)
• Round 3: Technical interview with the engineering team
• Round 4: Culture fit conversation with founders

Position Details:

• Start Date: Flexible, immediate joining preferred
• Location: Hyderabad, India (On-site)
• Employment Type: Full-time, permanent position

Surviant is an equal opportunity employer committed to diversity and inclusion. We welcome applications from all qualified candidates regardless of background, education, gender, or experience level.

Questions?
Reach out to our team at

Connect with us:

• Website:
• LinkedIn: @Surviant

Application Deadline: Open until filled Position Type: Full-Time