Staff AI Engineer

It's fun to work in a company where people truly BELIEVE in what they're doing!

We're committed to bringing passion and customer focus to the business.

Corporate Overview
Proofpoint is a leading cybersecurity company protecting organizations’ greatest assets and biggest risks: vulnerabilities in people. With an integrated suite of cloud-based solutions, Proofpoint helps companies around the world stop targeted threats, safeguard their data, and make their users more resilient against cyber-attacks. Leading organizations of all sizes, including more than half of the Fortune 1000, rely on Proofpoint for people-centric security and compliance solutions mitigating their most critical risks across email, the cloud, social media, and the web.
We are singularly devoted to helping our customers protect their greatest assets and biggest security risk: their people. That’s why we’re a leader in next-generation cybersecurity.
Protection Starts with People.  Proofpoint.
Staff AI Engineer

At Proofpoint, you will find an environment where innovation comes from creative collaboration and where everyone is empowered to go fast and have fun. If you find joy in taking the “im” out of impossible, you will feel right at home with our team! As an AI Engineer, you will be part of a high-caliber Engineering team responsible for designing, building, and managing AI-driven solutions to classify digital communications for regulated industries. You will work closely with Product Management and Engineering Leadership to define technical directions and architectural requirements. You will serve as a technical leader and mentor, promoting best practices in AI experimentation, model evaluation, and cost-effective implementation strategies.

Responsibilities

• Establish a clear AI/ML vision, build support across cross-functional teams, and drive high-impact AI initiatives to completion.

• Develop and deploy transformer-based models for classification tasks, leveraging pre-trained models, fine-tuning techniques, and embedding-based retrieval.

• Evaluate generalized/base models for low true-positive classification problems and design methods to improve performance.

• Implement prompt-based classification techniques and vector-based retrieval to improve accuracy and efficiency.

• Design and conduct experiments to measure model efficacy, balancing performance with computational costs.

• Research available AI/ML technologies and help make informed build vs. buy decisions.

• Ensure AI solutions are scalable, secure, and aligned with enterprise best practices.

• Stay current with the latest advancements in transformers, LLMs, and AI-driven text classification.

Requirements

• 10+ years of experience in software engineering, with a strong emphasis on AI/ML.

• 4+ years of hands-on experience working with transformer models or NLP Experience with one or more of the following AI/ML frameworks and libraries is desired: TensorFlow, PyTorch, and Hugging Face Transformers Expertise in selecting and applying transformer models for text classification problems.

• Experience with prompt engineering, vector-based retrieval, and embedding models.

• Strong understanding of AI cost structures, including model inference costs and optimization strategies.

• Previous experience designing and implementing experiments to measure model efficacy.

• Strong background in Cloud, Containers, Java and Linux-based architectures.

• Well-versed in Agile/Scrum development processes.

• Excellent communication and problem-solving skills. Degree in Computer Science, AI/ML, or a related field.

• Preferred/Desired Experience Experience working with Azure Machine Learning API and AWS Bedrock is a plus.

• Familiarity with LLMOps and deploying models in production environments. Understanding of data pipelines and distributed AI processing.
  Why Proofpoint
  Protecting people is at the heart of our award-winning lineup of cybersecurity solutions, and the people who work here are the key to our success.  We’re a customer-focused and a driven-to-win organization with leading-edge products. We are an inclusive, diverse, multinational company that believes in culture fit, but more importantly ‘culture-add’, and we strongly encourage people from all walks of life to apply.
  We believe in hiring the best and the brightest to help cultivate our culture of collaboration and appreciation. Apply today and explore your future at Proofpoint! #LifeAtProofpoint

#LI-PH1

If you like wild growth and working with happy, enthusiastic over-achievers, you'll enjoy your career with us!