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.
- 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.
- 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
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.