MiAI Law

Search & Ranking Optimization Engineer (Legal AI)
This role is based on-site in Tokyo, Japan and will be working directly with Fred Almeida.

About MiAI Law

MiAI Law is building the future of AI-powered legal research, delivering deep NLP models and AI-driven search technology dedicated to bridging the technology gap for lawyers by delivering advanced legal research tools and custom AI solutions. The company’s mission is to empower lawyers to provide efficient, affordable, and high-quality legal services while upholding data privacy and security standards. MiAI Law offers bespoke solutions tailored to individual client needs and scalable options for diverse requirements to enhance efficiency and productivity in the legal sector. We are looking for a Search & Ranking Optimization Engineer to develop scalable legal search infrastructure, optimize ranking models, and enhance legal document retrieval.

The Role

As a Search & Ranking Engineer at MiAI Law, you will:
  • Optimize multi-metric ranking algorithms for case law, contracts, and regulatory documents.
  • Implement scalable search infrastructure, integrating vector search, semantic retrieval, and metadata indexing.
  • Enhance search relevance using machine learning & NLP-based ranking models.
  • Work with legal AI researchers to integrate ML-driven ranking approaches.
  • Ensure high-performance legal search capabilities for large-scale legal datasets.

Key Responsibilities

Search Ranking Optimization
  • Design multi-metric ranking algorithms prioritizing court hierarchy, case recency, and legal significance.
  • Improve legal document search relevance using BM25, Elasticsearch, and Pinecone for vector search.
  • Optimize query expansion, auto-suggestion, and legal taxonomy integration.
Machine Learning & NLP for Search
  • Develop NLP-driven ranking models integrating legal semantics.
  • Implement BERT-based rerankers for legal search queries.
  • Apply vector embeddings & knowledge graph retrieval for enhanced case law search.
Performance Scaling & Infrastructure
  • Scale legal search workloads using distributed search engines (Elasticsearch, Solr, Pinecone).
  • Improve query latency, caching strategies, and search indexing efficiency.
  • Develop search monitoring tools to analyze query success rates & retrieval performance.

Requirements: Who You Are

Technical Skills:
  • 3+ years of experience in search, ranking, and information retrieval.
  • Strong background in search algorithms, Elasticsearch, Solr, Pinecone, or Vespa.
  • Proficiency in vector search, dense retrieval, and semantic search modeling.
  • Experience with ranking ML models (LTR, BERT-based rerankers, query expansion).
  • Knowledge of legal document retrieval, taxonomy classification, and NLP-driven legal search.

Why Join MiAI Law?

  • Be a key player in shaping the future of AI-powered legal search.
  • Collaborate with top engineers & legal professionals in Tokyo.
  • Work at the intersection of legal informatics, AI search ranking, and document retrieval.
  • Solve complex search optimization problems that impact real-world legal professionals.
  • Be part of an exciting, innovative Legaltech AI start-up that values collaboration and initiative.
  • Opportunity to work closely with senior leadership and contribute to the success of MiAI and key projects.

HOW TO APPLY

Please submit your application using the form below, and be sure to include:
  • Your resume
  • A link to your GitHub profile
  • And a brief outline of how your skills and experience align with this role
We look forward to your application!

Our Culture

At MiAI Law, our culture is the foundation of our success. We’re a high-performance, 
innovation-driven startup where every team member feels valued and empowered.

Join Us

Are you passionate about legal tech, excited by challenges, and ready to make a real impact?
If so, MiAI Law could be the place for you to thrive.

Submit your application

    Security validation*
    MiAI Law is committed to protecting your information. Your information will be used in accordance with the applicable data privacy law, our internal policies and our privacy policy. As MiAI Law is a global organisation, your information may be stored and processed by MiAI Law and its affiliates in countries outside your country of residence, but wherever your information is processed, we will handle it with the same care and respect for your privacy.