Software Engineer, AI

Great products need great architecture. As an engineer with the DiscoAI and Data Science teams, you’ll help lead, design and build the software and architecture that scales the business and can be confidently relied on. We are hiring people that approach design from a systems perspective and aspire to craft their systems well. A great candidate can deliver real customer value while pursuing architecture quality in availability, scalability, interoperability, modifiability, performance, security, and testability.

Disco is seeking a talented Software Engineer with 5+ years of experience to join our AI/ML team supporting our cutting edge AI features in the domain of legal technology. The AI/ML team designs, develops, and maintains Disco’s AI applications. This position will help shape our data infrastructure and how our data scientists and other AI engineers find, access and transform data. You will further develop experience in advanced AI/ML, working with small teams to assist with a range of tasks, from transitioning R&D prototypes to production-grade software to developing ML and data infrastructure to support rapid prototyping.

As a software engineer, you will work with ML Scientists to prepare the data they need and write software to help speed up their work. From initial exploratory research, through optimization all the way to production. You will work to reproduce any production issues and use debuggers to do root cause analysis. You will be involved in supporting our highly distributed systems and our performant ML platforms. This is not a direct production role, but we take pride in what we do and stand behind any software we deploy.

In the AI lab we look at solutions from a combined software and data perspective. We include a view of the data as the foundation of our solutions. As such, you will develop a deep understanding of data transformations, data distribution shifts, data clustering, anomaly detection and more. The proper data platform allows optimal hardware utilization and strikes the right balance between data latency, throughput and cost. 

Most of our applications are based on deep learning and statistical models, with particular emphasis on NLP tasks using techniques such as transfer learning with BERT. The successful candidate will have a conceptual understanding of machine learning workflows, ideally including experience implementing these workflows. AI/ML forms a core part of Disco’s brand and vision, and this position provides an opportunity to implement technology that will transform the legal domain, with significant benefits for the broader society.

• 5+ years experience in software development and deployment
• 2+ years of experience in Python
• 2+ years of experience in AWS
• 2+ years working with highly distributed systems
• 1+ years of experience working with unstructured data (NLP/NLU preferred)
• Experience with data management systems, including relational databases, data frames, and distributed file systems
• Reasonable prior exposure to best practices in software development lifecycle, and a commitment to implementing these practices appropriately in context
• Personal communication skills to facilitate the productive professional interaction with ML scientists and other engineers

• Conceptual knowledge of machine learning techniques and workflows
• Basic understanding of machine learning methods, especially deep learning for NLP
• Familiarity with a few 3rd party tools and libraries in the ML space, such as PyTorch, Pandas, Tensorboard, CometML, CUDA, cuDF, Dask, RAPIDS, Jupyter Labs/notebooks, etc.
• Continuous Integration/Continuous Deployment (CI/CD) and Infrastructure as Code (IaC)
• AWS offerings such as: batch, Sagemaker, SQS, ECR, ECS, Glue, Athena, etc.
• Kafka, protobuf, gRPC

• Open, inclusive, and fun environment
• Benefits, including medical, dental and vision insurance, as well as 401(k) 
• Competitive salary plus RSUs
• Flexible PTO 
• Opportunity to be a part of a company that is revolutionizing the legal industry
• Growth opportunities throughout the company