Senior/Staff Software Engineer - AI

Key Skills

AWS, Azure, ChatGPT, Cloud computing, Continuous Integration & Continuous Delivery - CI/CD, DataDog, Deep Learning, Docker, Git, Google Cloud Platform (GCP), Large scale data processing, Machine learning techniques, Natural Language Processing (NLP), NumPy, Pandas, Python Programming, PyTorch, REST API, TensorFlow

Job Description

Ironclad is looking for a Senior AI Engineer to join our team and help revolutionize the way legal teams handle contracts. As a Senior AI Engineer at Ironclad, you will have the opportunity to work with the latest cutting-edge technologies, including the HuggingFace framework, the OpenAI APIs, and more. You will leverage these to build state-of-the-art models and services that extract valuable information from contracts and provide insights to our customers.

You will work closely with other teams, including design and product, to build new features in our app that enable our customers to handle contracts more efficiently and effectively. You will also collaborate with data engineering to build datasets that are used to train machine learning models.

As a Senior AI Engineer, you will play a critical role in building an AI platform that can serve fast predictions and scale to support high throughput and multiple models. You will explore the latest models and AI techniques, including generative AI, question and answering, visual understanding, and more.

At Ironclad, we are on a mission to power the world’s contracts, and you will play a pivotal role in making that mission a reality. If you are an experienced AI Engineer with strong coding fundamentals and a passion for working with cutting-edge technologies, we encourage you to join us on this exciting journey. We work in a highly collaborative environment and strive to foster a positive, inclusive culture. Join us and help us build the product that legal teams love!

What you'll be doing:

• Work with product, data and engineering teams to build models and services that can ingest legal documents, extract key information and surface insights from these documents to our customers.
• Drive the technical vision for the team and leverage open source technologies as well as cloud services to build an enterprise grade machine learning platform.
• Lead the design and implementation of backend services that can ingest legal documents and extract relevant information that our customers care about.
• Drive the evolution of our ML platform, including building tooling to support model training, evaluation, inference serving, monitoring and alerting.
• Work with our data engineering and product teams to identify training data sources, ETL pipelines, and labeling tasks of interest. 
• Use the latest ML frameworks and open source tools to develop new model training pipelines
• Provide technical leadership for the engineering organization
• Partner with the product teams to innovate on user experience and solve business problems by making AI accessible through APIs
• Participate in interviews to hire new engineers into the team
• Mentor and train other junior engineers
• Maintain all software built by being available to fix P0s and bugs when users report

Required skills:

• BS Degree in Computer Science or related field
• Required (5+ years working with): Backend services, REST APIs, and Python
• Required (2+ years): NLP (text classification, entity extraction, Q&A, document analysis)
• ML model training, evaluation & serving
• Deep learning frameworks (pytorch, tensorflow)
• Data processing & manipulation with Python (pandas, numpy)
• Cloud services (azure, gcp, aws)
• Git or equivalent VCS

Nice to have:

• GPT or generative AI experience
• VertexAI
• Docker
• PDF and DOCX file format
• Using & configuring CI/CD systems
• Datadog or equivalent
• Exemplary coding practices with great focus on quality
• Passion for building and shipping products that customers love.
• Strong communication and collaboration skills.

Benefits:

• Health, dental, and vision insurance
• 401k
• Wellness reimbursement
• Take what you need vacation policy
• Generous parental leave for both primary and secondary caregivers

Base Salary Range: $140,000 - $252,000

The base salary range represents the minimum and maximum of the salary range for this position based at our San Francisco headquarters. The actual base salary offered for this position will depend on numerous factors, including individual proficiency, anticipated performance, and the location of the selected candidate. Our base salary is just one component of Ironclad’s competitive total rewards package, which also includes equity awards (a new hire grant, along with opportunities for additional awards throughout your tenure), competitive health and wellness benefits, and a commitment to career growth and development.