New opportunity

AI/ML Engineer

Blueflame Labs · Pune, India

About this role

Blueflame Labs is seeking an AI/ML Engineer in Pune to build and deploy scalable AI solutions for business and client needs. The role focuses on predictive analytics and deep learning and involves collaboration with data scientists, software engineers and product managers.

Responsibilities

  • Design and implement machine learning models and algorithms to address complex business problems; integrate AI/ML solutions into production systems.
  • Preprocess and analyze large datasets using Databricks and Python, extract insights, and ensure data quality for model training.
  • Build front-end interfaces for AI-driven applications with ReactJS. Train, optimize and fine-tune models with PyTorch for accuracy, performance and scalability.
  • Evaluate models through experiments, iterate on improvements, maintain scalable code, and document processes, models and results for reproducibility. Keep current with advances in machine learning, predictive analytics and deep learning.

Required qualifications

  • Bachelor’s or Master’s degree in Computer Science, Data Science, Mathematics or a related field, plus at least 3 years of experience as an ML developer or in a related AI/ML engineering role.
  • Strong Python proficiency; expertise in machine learning, PyTorch and deep learning techniques such as CNNs, RNNs and Transformers; hands-on Databricks experience for data processing, model development and deployment; and ReactJS proficiency.
  • Familiarity with Pandas, NumPy, Hadoop and Spark; understanding of software engineering practices including Git and CI/CD; strong problem-solving, collaboration and communication skills, including explaining technical concepts to non-technical stakeholders.

Preferred qualifications

  • A Ph.D.; knowledge of AWS, Azure or GCP for model deployment; experience with NLP, computer vision or reinforcement learning; familiarity with MLOps tools such as Kubeflow and MLflow and container technologies such as Docker and Kubernetes; or open-source ML contributions or relevant publications.

For full details about this role, click “Apply on company site” to view the employer’s original posting.

Skills for this role

Machine LearningPredictive AnalyticsDeep LearningPythonDatabricksReactJSPyTorchData PreprocessingModel EvaluationCNNsRNNsTransformersPandasNumPyHadoopSparkAWSAzureGCPGitCI/CDNatural Language ProcessingComputer VisionReinforcement LearningMLOpsKubeflowMLflowDockerKubernetesProblem Solving

Your skill match

Checking your profile…

YOUR NEXT STEP

Get interview-ready for this role

A focused preparation guide, built around this job’s responsibilities and requirements.

✦ AI-generated guide
Preparation suggestions, not the employer’s actual interview questions. Always check the original posting for current requirements.
01

Understand the role

What matters most

  • Predictive and deep-learning solutions

    Design and implement models for complex business problems, with a focus on predictive analytics and deep learning. The posting specifically calls for Python, PyTorch, and expertise in techniques such as CNNs, RNNs, and Transformers.

  • Data quality at scale

    Preprocess and analyze large datasets, extract insights, and ensure training-data quality. Hands-on Databricks experience is required for data processing, model development, and deployment.

  • Production integration and user experience

    Work with data scientists, software engineers, and product managers to integrate AI/ML solutions into production systems, and build ReactJS interfaces for AI-driven applications.

  • Evaluation and maintainability

    Run experiments, improve model performance, accuracy, and scalability, and contribute maintainable code. Document processes, models, and results for transparency and reproducibility.

Likely challenges to think through

  • Balancing model quality with operational constraints

    A likely trade-off is improving predictive accuracy without making a model too slow or difficult to scale in production. Prepare to explain how you would compare experiments, measure performance, and decide when an improvement merits deployment; this is a preparation hypothesis, not a reported problem at Blueflame Labs.

  • Connecting the data, model, and interface

    The stated Databricks, PyTorch, production-integration, and ReactJS duties span several layers. A likely difficulty is keeping data assumptions, model outputs, and the user-facing experience consistent. Prepare to walk through the interfaces and checks you would use across those layers; no particular architecture is specified.

  • Making experiments understandable to collaborators

    The role requires reproducible documentation and communication with non-technical stakeholders. A likely challenge is conveying model limitations and results clearly while retaining enough technical detail for others to reproduce the work. Prepare a concise explanation of an evaluation approach, its caveats, and the decisions it supports.

03

Practice your answers

Technical practice

How would you frame a business problem as a predictive analytics task?

What to coverSuggested preparation: Explain how you would define the target, choose a baseline, and select metrics that reflect the business objective. Discuss data leakage and the trade-off between model complexity and usefulness.

When would you choose a CNN, RNN, or Transformer?

What to coverSuggested preparation: Compare the kinds of data and relationships each architecture can model. Explain how data volume, computational cost, and performance would affect your choice.

How would you prepare and check a large training dataset in Databricks?

What to coverSuggested preparation: Describe checks for missing values, inconsistent records, and training-data leakage before modeling. Explain when distributed processing with Spark is useful compared with local Python tools such as Pandas and NumPy.

How would you train and fine-tune a deep learning model in PyTorch?

What to coverSuggested preparation: Walk through a training loop, validation, and choices such as learning rate and batch size. Discuss how you would diagnose overfitting while balancing accuracy, runtime, and scalability.

How would you design experiments to evaluate and improve a model?

What to coverSuggested preparation: Explain how you would establish a baseline, keep comparisons fair, and examine performance beyond a single aggregate metric. Describe what evidence would justify an iteration or show that a change did not help.

How would you integrate a trained model into a production system?

What to coverSuggested preparation: Describe the interface between model inference and the surrounding software, including input validation and failure handling. Explain trade-offs involving latency, scaling, and updating the model; cloud deployment is a plus in the posting, not a requirement.

How would you build a ReactJS interface for an AI-driven application?

What to coverSuggested preparation: Explain how the interface would submit inputs, display predictions, and handle loading or error states. Discuss how you would communicate uncertainty or limitations without making the result difficult to use.

How would you diagnose a model that performs well in development but poorly after deployment?

What to coverSuggested preparation: Compare production inputs and outcomes with the training and evaluation data, checking for data-quality or distribution changes. Explain how you would isolate a model issue from an integration issue before choosing a remedy.

How would you keep an ML codebase maintainable as it grows?

What to coverSuggested preparation: Discuss modular code, tests, Git practices, and CI/CD checks relevant to model and application changes. Explain how these practices reduce integration risk when working with data scientists and software engineers.

What would you document so an ML experiment can be reproduced?

What to coverSuggested preparation: Cover data preparation, model configuration, evaluation method, and results. The posting names MLflow and Kubeflow as preferred-tool examples; mention them only if useful to explaining how you would track experiments.

Behavioural practice

Tell me about a time you coordinated with different disciplines to deliver an ML feature.

What to coverSuggested preparation: Use a real situation involving relevant collaborators, the actions you took to align responsibilities, and the result. Make clear how you resolved differing technical or product priorities.

Tell me about a time you explained a model's results to a non-technical stakeholder.

What to coverSuggested preparation: Describe the audience's decision, how you translated technical findings or limitations, and what happened afterward. Focus on clarity rather than implying the model was more certain than the evidence supported.

Tell me about a time a data or model problem disrupted your work.

What to coverSuggested preparation: Give a real situation, the steps you took to identify the cause, and the result. Distinguish your own actions from the team's and explain what the evidence showed.

Tell me about a time you had to prioritize competing work in a fast-paced project.

What to coverSuggested preparation: Set out the situation, how you weighed urgency against quality or technical risk, and the outcome. Explain how you communicated the trade-off to collaborators.

Tell me about a time documentation or a handoff helped others use your work.

What to coverSuggested preparation: Choose a real example and describe what you documented, who needed it, and the result. Connect your actions to reproducibility or maintainability without inventing a formal process.

02

Sharpen your knowledge

Topics to prepare

  • Predictive analytics and model evaluation

    Conceptual revision: For a hypothetical business prediction problem, choose a baseline, explain suitable evaluation metrics and how you would detect overfitting. Relevant because the role designs predictive models and evaluates experiments to guide improvements.

  • Deep learning architecture choices

    Conceptual revision: Compare when a CNN, RNN or Transformer would fit a problem, including accuracy, compute and scalability trade-offs. The posting explicitly names these techniques and asks for deep-learning expertise.

  • Data quality and large-scale processing

    Conceptual revision: Explain how missing values, duplicates and inconsistent inputs can affect training, and how you would validate data when processing large datasets. Relevant to the stated Databricks and Python preprocessing responsibility.

  • Production integration and reproducibility

    Conceptual revision: Sketch how a trained model could be integrated into an application, then identify what code, experiment results and model details need documenting. The role calls for production integration, maintainable code and reproducible results.

Skills to revise

  • Python and PyTorch model development

    Hands-on exercise: Implement a small predictive model in Python and PyTorch, train it, tune one parameter and compare evaluation results. Both tools are must-haves, and the role involves training and optimizing models.

  • Databricks data-to-model workflow

    Hands-on exercise: Rehearse processing a dataset, checking its quality and preparing it for model development in Databricks; explain how the resulting model would move toward deployment. Hands-on Databricks use across processing, development and deployment is explicitly required.

  • ReactJS interface development

    Hands-on exercise: Build a simple ReactJS interface that accepts inputs and displays a hypothetical model prediction and its context. ReactJS proficiency for AI/ML application front ends is a must-have.

  • Experiment tracking and documentation

    Hands-on exercise: Run two model experiments, record their settings and metrics, and write a brief account of the result and next iteration. The posting asks for performance experiments, iteration and documentation for reproducibility; no specific tracking tool is required.

  • Git and CI/CD fundamentals

    Hands-on exercise: Put a small ML codebase under Git, add a basic automated check and explain how the check could fit into a CI/CD pipeline. The posting explicitly asks for an understanding of Git and CI/CD, alongside maintainable code.

  • Explaining results across teams

    Hands-on exercise: Give a two-minute, non-technical explanation of a model’s purpose, evaluation result and limitation, then describe what you would clarify with a product manager or software engineer. This reflects the stated collaboration and communication responsibilities.

04

Your preparation checklist

  • Map your evidence to the must-haves

    Prepare a one-page list of specific examples you can discuss for Python, PyTorch, Databricks, ReactJS, predictive analytics and deep learning. The posting identifies the first four as must-haves; do not imply experience you cannot substantiate.

  • Rehearse an end-to-end ML project walkthrough

    Choose a real project you can discuss and outline the problem, dataset, preprocessing, model choice, experiments, results and limitations. Highlight decisions relevant to the role’s focus on predictive analytics and deep learning.

  • Prepare a data-processing walkthrough

    Be ready to explain how you used Python and, if applicable, Databricks to process a large dataset and check training-data quality. Bring a concise account of the checks, insights and trade-offs rather than just a tool list.

  • Practice a PyTorch design exercise

    Work through a sample prediction problem: choose a suitable model, describe training and evaluation, and explain how you would improve accuracy, performance or scalability. Be ready to justify when a CNN, RNN or Transformer would be appropriate.

  • Sketch a production AI application

    Draw a simple design showing model integration with a production system and a ReactJS interface. Label it as a preparation exercise, and explain maintainability, Git and CI/CD considerations without presenting the design as Blueflame Labs’ architecture.

  • Prepare two versions of a technical explanation

    Explain one model result first to an engineer and then to a non-technical product or client stakeholder. Practice stating what the result means, its limitations and the next experiment; the role calls for cross-functional collaboration and clear communication.

  • Write focused questions for the employer

    Prepare questions about the AI problems the team prioritizes, how it evaluates deployed models, how Databricks and ReactJS fit into the workflow, and what documentation supports reproducibility. Treat these as questions, not assumptions about existing processes.

AI-generated from the job information · Prepared Sep 30, 2026 (UTC)Useful preparation, not a guarantee of eligibility or an offer.

Availability last checked: 5 Nov 2026. Status can change after a check; confirm on the employer’s website. Report an issue · Our checking policy

KEEP EXPLORING

Similar AI jobs

Related skills and specializations in India. Matched to this role, not your profile.

Explore more jobs
Finding similar opportunities…