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:

Machine Learning Engineer

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 🚀