Junior AI Engineer

at  Proofpoint

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.
About the Team
Proofpoint Labs is an internal machine learning group that consults across Proofpoint’s entire product portfolio. We are a group of data scientists and engineers who love keeping up with the latest ML research, fostering a collaborative and creative workplace, and solving challenging problems. Over the past several years, we have developed data-driven product features, leveraging a range of model architectures from tree-based models to state-of-the-art transformers. The projects have spanned the full range of modern ML techniques (deep and classical learning), approaches (supervised and unsupervised learning) and data types (textual, image, and structured data). Each new project surfaces novel puzzles to be solved, while production deployments to our cloud ML Platform expose new MLOps challenges. Recently, our organization began spearheading company-wide initiatives around large language models and their application to the business by creating resources, putting together project teams, and organizing hack days.

We are a growing team with many opportunities for ownership and impact, guided by our four values:

• Human Beings, Not Human Doings: We bring our whole self into every part of our work. We possess worth and value beyond our code.
• Humble But Confident: We are confident in our current abilities while celebrating the strengths of our team and our own areas for growth.
• Openly Collaborative: We strive for pristine documentation, transparent chat rooms, and realistic management of tech debt and tribal knowledge.
• Impactful Innovators: We seek and scope meaningful work that we approach creatively but efficiently.

What You’ll Do:

• Write Python code to facilitate all components of the ML lifecycle.
• Develop appropriate unit and integration tests.
• Write Terraform code to provision parts of the ML infrastructure in AWS.
• Measure and monitor system performance for scalability and up-time.
• Build and support LLM applications that interface with proprietary internal data sources.
• Work with data scientists to capture and label data, build and deploy ML models, and measure their efficacy.
• Interact with other Proofpoint teams on product integration.
• Present plans and results within the team and externally to non-technical stakeholders.

What You Need:

• BS and/or MS degree in Computer Science or closely related field.
• Excellent written and verbal communication skills.
• Passion for learning new skills and technologies.

What Will Help:

• Previous professional experience at a SaaS company.

Experience with Python, Unix/Linux shell, Git, and infrastructure-as-code in a cloud environment (AWS preferred).

• Experience with containerization, cluster management, and orchestration (Kubernetes preferred)
• Experience with LLMs.
• Experience with AWS SageMaker and other AWS technologies.
• Interest in data processing and machine learning.
• Interest in computer, network, or email security.

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

• Write Python code to facilitate all components of the ML lifecycle.
• Develop appropriate unit and integration tests.
• Write Terraform code to provision parts of the ML infrastructure in AWS.
• Measure and monitor system performance for scalability and up-time.
• Build and support LLM applications that interface with proprietary internal data sources.
• Work with data scientists to capture and label data, build and deploy ML models, and measure their efficacy.
• Interact with other Proofpoint teams on product integration.
• Present plans and results within the team and externally to non-technical stakeholders