Lead Machine Learning Engineer (NLP) (Remote India)

Draft (formerly Contentfly) is building the content infrastructure that companies leverage to build, scale, and optimize their revenue engines. We're a fully remote and distributed team of high performers on a mission to help companies tell their stories. We're a Series A company backed by Khosla Ventures, YCombinator & other top-tier investors, we've grown to have an incredibly talented team of 50+ people and $8m in ARR in the last year and a half. With a world-class talent marketplace and a team that consistently raises the bar, we have an opportunity to build an all-in-one platform for exceptional content marketing.

Reporting to the VP of Engineering, this position will play a key role in defining the AI strategy at the company. You will be working closely with product and engineering teams to design, develop and deploy ML models at scale. You will have a measurable impact on the growing revenue of the company along with high visibility across the engineering team.

You are:

• An effective communicator, you have the charisma to command attention, strategize and problem-solve on your feet when needed with internal and external team members
• A Self-Starter, you know your craft but are still eager to improve your skillset in the ever-evolving space of Machine Learning.
• Ownership , you have a strong sense of responsibility and accountability for the tasks/projects you work on.
• Leadership , manage people, situations, and items effectively to shape and guide the growth of AI at Draft.

Key Responsibilities

Here's what you'll be doing day to day...

• Research: Stay up-to-date with the latest research in NLP/NLU space.
• Design: Participate in product design discussions and decisions
• Implement: Develop and implement AI software/products aligned with the growing demands of our customers. Test, debug, support and enhance software as per product specifications. Participate in swift iteration cycles related to Training/Finetuning + Testing + Deployment of NLP models
• Collaborate: Work with various teams to establish organizational best practices that scale with our team and products, consistently resulting in high-quality software systems that set industry standards.
• Mentor : You shall mentor, support, and lead a small team of NLP Engineers

Your Experience and Intangible Traits:

The ideal candidate for the Lead Machine Learning Engineer position will possess the following competencies:

• Ability to understand business problems and formulate them as Machine Learning problems
• A Research-to-Implementation Mindset: ability to spearhead a project from ideation to deployment in production
• Strong understanding of Ranking/Recommendation systems
  • 1+ year of experience working with Recommendation/Ranking systems
  • Knowledge of traditional and Deep Learning based Ranking/Recommendations systems
  • Experience working with semi-structured data
• 2+ years of experience working with NLP/NLU based technologies
• Strong understanding of NLP/NLU fundamentals
  • Knowledge of RNN, GRU, LSTM, and Transformer based models
  • Expertise in traditional and state-of-the-art text classification and generation models
  • Understanding of language models like BERT, GPT3, T5, etc.
• Dealing with structured and semi-structured data for hypothesis testing, multivariate analysis, model explainability, clustering, etc using statistical and machine learning based algorithms
• Rapid prototyping using open-source algorithms and shaping them to meet product needs
• Exceptionally strong Python skills - Minimum 5 years of experience
• Experience deploying ML models in the Cloud - At least 3 models deployed in the past
• Prior experience with AWS and related services
• Experience with databases (Postgres/MySQL)
• ML framework: Pytorch / Tensorflow
  • Experience with experiment tracking systems
  • Docker
  • Bonus: Experience with MLOps - cortex.dev
• Experience managing Machine Learning Engineers
• Good to have: previous open source contributions to NLP projects/libraries/packages

Why you'll love working here:

• Unlimited vacation/PTO + Stock Options
• Fully remote and distributed teams
• Full Health benefits
• Paid Parental Leave
• Free laptop and home office budget, we've also got your cellphone/internet covered too!
• Ongoing training and education budgets
• We'll fly you around the world multiple times a year to hang out with the Draft Picks in interesting places

Like what you see and think you’re a perfect fit? Apply today.

Please be assured we'll treat any information you share with us with the utmost care, only use your information for recruitment purposes and will never sell it to other companies for marketing purposes. Please review our privacy policyfor further information.

Draft is proud to be an equal opportunity employer, we are committed to fostering an inclusive, accessible environment, where all employees feel valued, respected and supported. We are dedicated to creating a culture with a diversity of talented individuals who join, stay, and work in an environment that enables them to thrive. Draft is committed to providing accommodations. If you require an accommodation, we will work with you to meet your needs.

#LI-Remote