Sr. AI Machine Learning Engineer

Responsibilities

• Design and implement end-to-end safety-focused frameworks for Large Language Models (LLMs)
• Develop risk mitigation techniques and safe inference strategies to ensure reliable AI behavior
• Identify system vulnerabilities through adversarial testing, bias detection, and mitigation strategies
• Collaborate with cross-functional teams to integrate safety mechanisms into AI workflows and pipelines
• Design and optimize LLM architectures to improve performance, scalability, and efficiency
• Fine-tune pre-trained LLMs on domain-specific datasets to enhance task performance
• Stay current with the latest research papers and advancements in deep learning
• Develop working prototypes from research ideas using strong software engineering skills

Requirements

• Bachelor's or Master's degree in Computer Science, Electrical Engineering, or a related field
• Proven experience developing and deploying LLMs, such as GPT, BERT, and their variants
• Strong programming proficiency in Python
• Hands-on experience with deep learning frameworks like TensorFlow or PyTorch
• Knowledge of distributed training techniques
• Experience working with cloud computing platforms such as AWS, Azure, or Google Cloud
• Excellent problem-solving, communication, and collaboration skills

About the Company

A10 Networks provides solutions that protect and optimize application delivery and security in multi-cloud environments.