Disco is seeking a talented Software Architect 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 and is offering an opportunity for an engineer to apply their experience in AI/ML while 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 and AI research.

Your Impact

Engineers on the AI/ML team are engaged in building distributed systems for ML ops, data preparation, and scale-up of R&D prototypes into production-ready systems. Most of our applications are based on deep learning, 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 professional 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. 

This position will also involve providing direct support to ML Scientists and Engineers by troubleshooting infrastructure and software issues within our highly distributed AI systems, improving hardware utilization, throughput and latency, hunting rare bugs observed in our systems, developing regression tests and telemetry to prevent bugs in the future, and implementing more robust software in response. This is not a direct production role, but we take pride in what we do and stand behind any software we deploy. While working with us you will employ processes, coding methodologies, architecture techniques, and testing strategies that improve robustness, maintainability, and reliability that will significantly strengthen your profile as a Software Architect or Machine Learning Engineer. 

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 within our sandbox and use debuggers to do root cause analysis. In the 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 an understanding of data transformations, data distribution shifts, data clustering, anomaly detection and more.

What You'll Do

  • Design, develop, debug, and deploy software to support AI/ML solutions in an R&D context and in backend production components
  • Architect, develop, and improve ML Ops infrastructure in support of ML Scientists
  • Harden research prototypes in preparation for production deployment
  • Provide engineering support for top-level ML scientists to help them access and manipulate data and build research prototypes
  • Advise ML scientists on the engineering implications and opportunities of their ideas
  • Adhere to appropriate engineering standards of scalability and robustness both in research and production-ready code 
  • Participate in domain-driven design processes for ML Ops infrastructure, from requirements gathering to system design, implementation, and deployment

Who You Are

  • 7-10+ years experience in software development and deployment
  • 2+ years of experience in AWS
  • Experience across the software development lifecycle, from design through development, deployment and maintenance
  • Experience with data management systems, including relational databases, data frames, and distributed file systems
  • Reasonable prior exposure to best practices in software development, 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

Even Better If You Have...

  • 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, etc.
  • Continuous Integration/Continuous Deployment (CI/CD) and Infrastructure as Code (IaC)

Technology Stack

  • Cloud Provider - AWS: EC2, Lambda, Aurora, Redshift, DynamoDB, ECS, SQS, SNS, Kinesis, S3, CloudFront, CloudFormation, SageMaker, KMS, CodePipeline, etc.
  • DSL-based Search: multiple large-scale Elasticsearch Clusters searched using our Disco Query Language (DQL)
  • Event Bus: Kafka and Schema Registry
  • 3rd Party Vendors: Redis, Auth0 for Cloud Identity Federation (SSO, SAML, etc)
  • AI: MinHash, FastText, Word2Vec, Convolution Neural Nets, Algorithmia (Lambda with GPUs) for training, PyTorch, Recurrent Neural Networks, Latent Dirichlet Allocation for Topic Modeling, etc.
  • Deployment: Terraform, Docker (via ECS), Consul for App Config, Service Discovery, Shared Secrets
  • Visibility: ELK Stack for logging, Datadog, New Relic, Sentry.io
  • Programming Languages: Kotlin (JVM), JavaScript, C#/.NET, Python
  • Transport Mechanisms: Protobuf, Avro, HTTP Rest/JSON
  • CI/CD: Jenkins, CodePipeline, GitHub, Artifactory


DISCO provides a cloud-native, artificial intelligence-powered legal solution that simplifies ediscovery, legal document review and case management for enterprises, law firms, legal services providers and governments. Our scalable, integrated solution enables legal departments to easily collect, process and review enterprise data that is relevant or potentially relevant to legal matters.

In 2020, 171 law firms in the 2020 AmLaw 200 used DISCO in the course of legal work on behalf of their clients. More than 800 enterprises, law firms, legal services providers and government organizations are DISCO customers.

Are you ready to revolutionize the practice of law? Join us!

Perks of DISCO

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

We are an equal opportunity employer and value diversity. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

Please note that DISCO has a mandatory COVID vaccination policy which requires all employees in the U.S. to be fully vaccinated, subject to applicable legal exemptions.


Apply for this Job

* Required

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in DISCO’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Form CC-305

OMB Control Number 1250-0005

Expires 05/31/2023

Voluntary Self-Identification of Disability

Why are you being asked to complete this form?

We are a federal contractor or subcontractor required by law to provide equal employment opportunity to qualified people with disabilities. We are also required to measure our progress toward having at least 7% of our workforce be individuals with disabilities. To do this, we must ask applicants and employees if they have a disability or have ever had a disability. Because a person may become disabled at any time, we ask all of our employees to update their information at least every five years.

Identifying yourself as an individual with a disability is voluntary, and we hope that you will choose to do so. Your answer will be maintained confidentially and not be seen by selecting officials or anyone else involved in making personnel decisions. Completing the form will not negatively impact you in any way, regardless of whether you have self-identified in the past. For more information about this form or the equal employment obligations of federal contractors under Section 503 of the Rehabilitation Act, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

You are considered to have a disability if you have a physical or mental impairment or medical condition that substantially limits a major life activity, or if you have a history or record of such an impairment or medical condition.

Disabilities include, but are not limited to:

  • Autism
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, or HIV/AIDS
  • Blind or low vision
  • Cancer
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or hard of hearing
  • Depression or anxiety
  • Diabetes
  • Epilepsy
  • Gastrointestinal disorders, for example, Crohn's Disease, or irritable bowel syndrome
  • Intellectual disability
  • Missing limbs or partially missing limbs
  • Nervous system condition for example, migraine headaches, Parkinson’s disease, or Multiple sclerosis (MS)
  • Psychiatric condition, for example, bipolar disorder, schizophrenia, PTSD, or major depression

1Section 503 of the Rehabilitation Act of 1973, as amended. For more information about this form or the equal employment obligations of Federal contractors, visit the U.S. Department of Labor's Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.