Machine Learning Engineer

Patronus AI, Inc.; New York, San Francisco (NY, CA); Hybrid; USD 90,000 - 150,000; Full time

Job summary

Join an innovative firm dedicated to revolutionizing AI oversight! As a Machine Learning Engineer, you'll tackle groundbreaking challenges in AI evaluation and model training. Collaborate with a talented team from top tech backgrounds to develop state-of-the-art systems that ensure AI aligns with human values. This role offers a unique opportunity to engage in cutting-edge research while contributing to scalable solutions in a dynamic environment. If you're passionate about AI and eager to make a significant impact, this position is perfect for you!

Benefits

Competitive salary and equity packages

Health, dental, and vision insurance

401k plan

Unlimited PTO

Fun global offsites

Qualifications

• Strong understanding of NLP and machine learning concepts.
• Experience with training language models and AI research.

Responsibilities

• Develop state-of-the-art systems for AI evaluation.
• Train language models and evaluate alignment with human preferences.
• Conduct novel research on language models.

Skills

Proactive mindset

Willingness to learn

Passion for research

Experience with NLP

Knowledge of machine learning concepts

Education

BS/MS in Computer Science

PhD in relevant field

Tools

PyTorch

Hugging Face Transformers

AWS

Docker

Kubernetes

Job description

Note: We have a preference for candidates that are able to work out of NYC or SF for this role!

About Patronus AI

Patronus AI’s mission is to provide the security and risk management layer for AI. We are solving the problem of scalable oversight - how can humans continue to supervise AI systems when AI far outperforms them in many real world scenarios? Our vision is a world in which AI evaluates AI. Our founding team comes from top applied ML and research backgrounds, including Facebook AI Research (FAIR), Airbnb, Meta Reality Labs, and quant finance. As a team, we have published research papers at top ML conferences (NeurIPS, EMNLP, ACL), designed and launched Airbnb’s first conversational AI assistant, pioneered causal inference at Meta Reality Labs, exited a quant hedge fund backed by Mark Cuban, and scaled 0→1 products at high growth startups. We are backed by Lightspeed Venture Partners and high profile operators like Amjad Masad, Gokul Rajaram, and Fortune 500 executives and board members. We are advised by Douwe Kiela, Adjunct Professor at Stanford University and former Head of Research at Hugging Face.

• Competitive salary and equity packages
• Health, dental, and vision insurance plans
• 401k plan
• Unlimited PTO
• Fun global offsites!

Responsibilities

As a Machine Learning Engineer at Patronus AI, you will solve the most important and challenging open research problems facing society’s adoption of AI today. In this role, you will:

• Develop state-of-the-art systems for AI evaluation.
• Train language models for novel use cases, such as evaluating whether content is engaging, hallucinatory, age appropriate, or contains PII.
• Evaluate whether language models are aligned with human preferences.
• Collect high quality, novel datasets for classification and generative tasks, through synthetic data augmentation techniques and publicly available datasets.
• Conduct novel research on red teaming language models.
• Keep up to date on current literature and latest technologies, and synthesize findings in research updates.
• Experiment with latest technologies and proactively suggest experiments and improvements to systems.
• Maintain a research journal and well organized results logs.
• Benchmark performance of different models by running robust experiments and ablations.
• Meet regularly with the CTO and research advisor to discuss progress updates and next steps.
• Develop efficient, scalable systems for hosting models for production use cases using AWS, Docker and Kubernetes.

Qualifications

“ The number one qualification to succeed in this machine learning course is gumption. ” - John Lafferty, CS Professor at Yale

Above all, we look for a proactive mindset, willingness to learn, relentless drive, and passion for research and engineering. You are a great fit if you have a background in the following:

• BS/MS in Computer Science, Mathematics, Statistics, or other quantitative field; PhD preferred
• Knowledge and understanding of state-of-the-art machine learning concepts, with a focus on NLP.
• Familiarity with transformer-based architectures, attention mechanisms, evaluation metrics and benchmarks.
• Experience training language models in applied or research settings. Bonus if you have overcome challenges in training LLMs, scalable inference, or putting a LLM into production!
• Deep familiarity with PyTorch and ML tooling such as job schedulers, the Hugging Face transformers library, and Weights & Biases.
• Experience conducting AI research in an academic or industry research lab. Bonus if you have published and reviewed papers in academic conferences (NeurIPS, ICML, ACL etc.).
• Ability to synthesize research papers and stay up-to-date on latest developments in research and open source.
• Have good character, integrity and respect for others!

Patronus AI is an equal opportunity employer. We celebrate diversity in our workplace, and all qualified applicants will receive consideration for employment without regard to age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or other legally protected characteristics.