
AI ML Data Engineer
KLA
We do not know your resume yet
Upload your resume to unlock your actual match score and identify important JD keywords before applying.
Recruiters may search these ATS Keywords in your resume
Keywords
Job Description
You will be part of a cutting-edge team working on Large Language Models (LLMs), Machine Learning, Deep Learning, and Retrieval-Augmented Generation (RAG) pipelines. Youll help design, build, and evaluate AI systems that solve complex real-world problems at scale.
Key Responsibilities
- Develop and optimize RAG pipelines: document chunking, embedding generation, vector storage, retrieval, reranking, and grounded generation with citations
- Work on LLM-based applications: fine-tuning open-source models (LLaMA, Mistral, etc), building prompt strategies, and deploying inference services
- Contribute to machine learning models (classification, regression, recommendation, anomaly detection) and deep learning architectures (CNNs, RNNs, Transformers)
- Implement robust model evaluation frameworks (accuracy, F1, BLEU, perplexity, hallucination detection, relevance)
- Collaborate with senior engineers on scalable pipelines, guardrails, and integration with enterprise systems
- Programming & Foundations
- Strong in Python, data structures, and algorithms
- Hands-on with NumPy, Pandas, Scikit-learn for ML prototyping
- Machine Learning
- Understanding of supervised/unsupervised learning, regularization, feature engineering, model selection, cross-validation, ensemble methods (XGBoost, LightGBM)
- Deep Learning
- Proficiency with PyTorch (preferred) or TensorFlow/Keras
- Knowledge of CNNs, RNNs, LSTMs, Transformers, Attention mechanisms
- Familiarity with optimization (Adam, SGD), dropout, batch norm
- LLMs & RAG
- Hugging Face Transformers (tokenizers, embeddings, model fine-tuning)
- Vector databases (Milvus, FAISS, Pinecone, ElasticSearch)
- Prompt engineering, function/tool calling, JSON schema outputs
- Data & Tools
- SQL fundamentals; exposure to data wrangling and pipelines
- Git/GitHub, Jupyter, basic Docker
Nice to Have
- Built a personal ML/LLM project (chatbot, RAG app, document Q&A, finetuned model)
- Familiarity with LangChain/LlamaIndex/Agno frameworks
- Knowledge of cloud platforms (Azure/AWS/GCP) and MLOps basics (CI/CD, MLflow, W&B)
- Exposure to knowledge graphs or multi-agent workflows
What Were Looking For
- Solid academic foundation with strong applied ML/DL exposure.
- Curiosity to learn cutting-edge AI and willingness to experiment.
- Clear communicator who can explain ML/LLM trade-offs simply.
- Strong problem-solving and ownership mindset.
Minimum Qualifications
- Doctorate (Academic) Degree and 0 years related work experience; Master's Level Degree and related work experience of 3 years; Bachelor's Level Degree and related work experience of 5 years
Swing Shift:2-11 pm
About The Company
KLA
KLA develops industry-leading equipment and services that enable innovation throughout the electronics industry. We provide advanced process control and process-enabling solutions for manufacturing wafers and reticles, integrated circuits, packaging and printed circuit boards. In close collaboration with leading customers across the globe, our expert teams of physicists, engineers, data scientists and problem-solvers design solutions that move the world forward. Visit us at: www.kla.com Statements made on LinkedIn may constitute forward-looking statements under federal securities laws. These forward-looking statements involve risks and uncertainties that could significantly affect the expected results and are based on certain key assumptions. Due to such uncertainties and risks, no assurances can be given that such expectations will prove to have been correct, and readers are cautioned not to place undue reliance on such forward-looking statements, which speak only as of the date indicated. Other risks that KLA faces include those detailed in KLA filings with the Securities and Exchange Commission, including KLA's annual report on Form 10-K and quarterly reports on Form 10-Q. Forward-looking statements made by third parties do not necessarily reflect the opinion of KLA, are outside of KLA’s control and have not been verified or otherwise vetted by KLA.
How to Apply Better for This Job
This section explains the correct next step without forcing sign-in immediately.
Check ATS score before applying
Scan your resume for ATS readability, formatting issues, missing sections, weak keywords, and content gaps.
Customize your resume for this JD
Match your resume with the job description and add ai , ml , git , sql , aws , keywords where they fit naturally.
Find similar jobs too
Do not depend on one opening. Use your resume to find similar frontend jobs across relevant job platforms.
Ready with your customized resume?
Once your resume includes the right skills and is ATS-friendly, you can apply directly on the source platform.
Market Insights:Best AI/ML Solution Specialist Jobs in India
Find the latest AI/ML Solution Specialist jobs across top Indian cities. Compare job counts by location and apply where hiring demand is higher.