Machine Learning Engineer

About RocketFrog.ai:

RocketFrog.ai is an AI Studio for Business, delivering deep science and real-world impact through cutting-edge AI solutions. We specialize in Agentic AI, Deep Learning, and full-stack AI-first product development across Healthcare, Pharma, BFSI, and Hi-Tech industries.

🚀 Ready to take a Rocket Leap with Science?

Role Overview:

We are seeking a high-potential Machine Learning Engineer with 2–4 years of experience and a strong academic background to join our AI innovation team. You’ll design and implement ML models that drive intelligent automation and AI-first products for enterprise-grade use cases.

Key Responsibilities:

• Design, develop, and optimize machine learning models for NLP, computer vision, or multi-modal applications
• Build scalable pipelines for data preprocessing, training, and evaluation
• Translate academic research and ideas into efficient, production-ready implementations
• Explore and research relevant papers to propose plausible approaches for solving business problems
• Apply optimization techniques such as quantization, pruning, and distillation
• Collaborate with product, engineering, and domain teams to align ML systems with business needs
• Stay updated with the latest developments in ML frameworks, tooling, and deployment strategies

Required Skills & Expertise:

• Strong knowledge of Linear Algebra, Probability, and Statistics
• Machine Learning Proficiency: Hands-on experience with supervised, unsupervised, and deep learning methods (CNNs, RNNs, Transformers)
• Model Training Workflows: Familiarity with training-validation pipelines, k-fold cross-validation, early stopping, and hyperparameter tuning
• Evaluation Metrics: Experience evaluating models using confusion matrix, PR curves, F1 score, accuracy, and ROC-AUC
• Frameworks: Proficient in PyTorch with strong coding practices (clean, modular, testable)
• Tools: Experience with model libraries and hubs such as Hugging Face Transformers, TorchVision, etc.
• Data Handling: Ability to manage large datasets, custom data loaders, and data augmentation workflows
• Deployment: Experience with model packaging (TorchScript, ONNX) and serving (FastAPI, TorchServe)
• Experimentation & Reproducibility: Familiarity with tools like Weights & Biases, MLflow, or equivalent

Desirable Skills:

• Experience with seq2seq models, LLM fine-tuning, or multi-modal architectures
• Hands-on with knowledge distillation and other model compression techniques
• Exposure to Agentic AI or orchestration frameworks like LangChain or LangGraph
• Understanding of vector databases, embedding stores, and RAG pipelines
• Contributions to open-source ML projects or academic research
• Cloud AI Platforms: Preference for experience with Amazon SageMaker, Azure AI Services, Google Vertex AI, or Azure AI Foundry

What We Look For:

• Strong foundations in mathematics, algorithmic thinking, and ML system design
• Curiosity-driven mindset and a passion for solving complex problems
• Ability to convert ideas into scalable and efficient solutions
• Desire to learn and grow in a high-performance, innovation-first environment
• Clear communication and collaborative problem-solving skills

Required Education Background:

• B.Tech / M.Tech / MS from IISc, IITs, Top NITs, BITS Pilani, IIIT-H, DTU, NSUT, or equivalent Tier-1 institutions.

Why Join RocketFrog.ai?

🚀 Be part of the growth journey of RocketFrog.ai!

• Work in an AI-driven company that is making a real business impact.
• Enjoy a dynamic, fast-paced environment with career growth opportunities.
• Apply now and be part of our journey to shape the future of AI 🚀