Job Description
About the Position
We are seeking an experienced AI/ML Engineer to join our team. As a key member, you will be responsible for designing, validating, and integrating cutting-edge machine learning models and algorithms.
Main Responsibilities
- Develop and implement machine learning models and algorithms.
- Collaborate with data scientists and other stakeholders to understand and define project goals.
- Maintain data infrastructure and ensure scalability and efficiency of data-related operations.
- Stay up-to-date with advancements in AI and propose their integration for operational enhancement.
- Communicate complex data findings in a clear and understandable manner to non-technical stakeholders.
- Ensure data privacy and security guidelines are upheld.
Qualifications
- A strong grasp of computer architecture, data structures, system software, and machine learning fundamentals.
- Theoretical understanding of machine learning.
- Experience with mapping NLP models (BERT and GPT) to accelerators and awareness of trade-offs across memory, bandwidth, and compute.
- Experience with Vector databases like Chroma Db, Pinecone, PGVector or similar.
- Experience with Large language Models like GPT 3.5, GPT 4, Llama, Gemini, Mistral etc.
- Experience in LLM integration framework like langchain, Llamaindex, AgentGPT etc.
- Experience with ML Models from definition to deployment, including training, quantization, sparsity, model preprocessing, and deployment.
- Proficiency in Python development in a Linux environment and using standard development tools.
- Experience with deep learning frameworks (such as PyTorch, Tensorflow, Keras, Spark).
- Working knowledge of Artificial Neural Networks (ANNs), Convolutional Neural Networks (CNNs), Recurrent Neural Networks (RNNs), and Generative Adversarial Networks (GANs).
- Experience in training, tuning, and deploying ML models for Computer Vision (e.g., ResNet), and/or Recommendation Systems (e.g., DLRM).
- Experience deploying ML workloads on distributed systems.
- A self-motivated team player with a strong sense of ownership and leadership.
- Strong verbal, written, and organizational skills for effective communication and documentation.
- A research background with a publication record.
- Work experience at a cloud provider or AI compute/sub-system company.
- Knowledge of cloud computing platforms and services, such as AWS, Azure, or Google Cloud.
- Experience with information security and secure development best practices.
- A Bachelor's or higher degree in Computer Science, Engineering, Mathematics, Statistics, Physics, or a related field.
- 2-4 yrs of hands-on experience in AI/ML.
Note
While the Job Highlights section is provided separately, this section is dedicated to the detailed job description.