AI/ML Engineer
Tata Consultancy Services
AI/ML Engineer
Key Responsibilities
- Design and implement scalable and efficient data architectures to support generative AI workflows.
- Fine tune and optimize large language models (LLM) for generative AI, conduct performance evaluation and benchmarking for LLMs and machine learning models.
- Apply prompt engineer techniques as required by the use case.
- Collaborate with research and development teams to build large language models for generative AI use cases, plan and breakdown of larger data science tasks to lower-level tasks.
- Lead junior data engineers on tasks such as design data pipelines, dataset creation, and deployment, use data visualization tools, machine learning techniques, natural language processing, feature engineering, deep learning, statistical modelling as required by the use case.
Requirements & Tech Stack
- Experience in working with various ML libraries and packages like Scikit learn, Numpy, Pandas, Tensorflow, Matplotlib, Caffe.
- Deep Learning Frameworks: PyTorch, spaCy, Keras.
- Deep Learning Architectures: LSTM, CNN, Self-Attention and Transformers.
- Experience in working with Image processing, computer vision is must.
- Designing data science applications, Large Language Models(LLM), Generative Pre-trained Transformers (GPT), generative AI techniques, Natural Language Processing (NLP), machine learning techniques.
- Programming experience in Python.
- Jupyter Notebook.
- Common data science packages (e.g., tensorflow, scikit-learn, keras).
- LangChain.
- Flask.
- FastAPI.
- Prompt engineering.
- Experience working with vectored databases and graph representation of documents.
- Experience with building or maintaining MLOps pipelines.
- Exposure to Docker, Kubernetes.
Qualifications
- Strong written and verbal communications.
- Excellent interpersonal and collaboration skills.
- Experience in Cloud computing infrastructures like AWS Sagemaker or Azure ML for implementing ML solutions is preferred.