Proofpoint is seeking an AI Engineer to join a high-caliber Engineering team. In this role, you will focus on designing, building, and supporting AI-driven solutions to classify digital communications for regulated industries. You will collaborate closely with Product Management and fellow engineers to translate requirements into practical, scalable solutions.
What You'll Do
- Contribute to the development and deployment of transformer-based models for classification tasks, leveraging pre-trained models, fine-tuning, and embedding-based retrieval.
- Evaluate generalized models for low true-positive classification problems and help improve model performance.
- Implement prompt-based classification techniques and vector-based retrieval approaches to enhance accuracy and efficiency.
- Design and execute experiments to measure model efficacy, balancing performance with computational cost.
- Collaborate with cross-functional teams to translate product requirements into practical AI/ML solutions.
- Support build vs. buy evaluations by researching and testing available AI/ML technologies.
- Help ensure AI solutions are scalable, secure, and aligned with enterprise best practices.
- Stay current with advancements in transformers, LLMs, and AI-driven text classification, applying relevant innovations where appropriate.
What We're Looking For
- 5+ years of experience in software engineering, with experience in AI/ML-focused projects.
- 2+ years of hands-on experience working with transformer models or NLP.
- Experience with one or more AI/ML frameworks such as TensorFlow, PyTorch, or Hugging Face Transformers.
- Practical experience applying transformer models to text classification problems.
- Familiarity with prompt engineering, vector-based retrieval, and embedding models.
- Understanding of AI cost considerations, including model inference costs and optimization techniques.
- Experience contributing to experiments to evaluate and improve model performance.
- Background in cloud environments, containers, Java, and Linux-based architectures.
- Familiarity with Agile/Scrum development processes.
- Strong problem-solving skills and the ability to communicate effectively within a team environment.
- Degree in Computer Science, AI/ML, or a related field.
Technical Stack
- TensorFlow
- PyTorch
- Hugging Face Transformers
- Java
- Linux
Team & Environment
You will be part of a high-caliber Engineering team, collaborating closely with Product Management.
Benefits & Compensation
- Competitive compensation
- Comprehensive benefits
- Flexible work environment
- Annual wellness and community outreach days
- Global collaboration and networking opportunities
Proofpoint is an equal opportunity employer.
