AI Engineer SDE-III

This is an advanced role distributed across research, engineering, and product. Relatedly, it is a combination of invention and optimization. These are the problem statements that this role will likely continue or begin our work on—

Constructing measures of search reliability/comprehensiveness and success/effectiveness; comparing ordered sets of search results; relevant notions of similarity and distance to evaluate IR

Implementing and automating benchmarks using labeled data and expert human feedback, interfacing with our legal research fellows

Measures of the popularity of a result for a class of queries, to enable reranking by observing users

Indexing algorithms and data structures, and infrastructure for billion-scale, multi-engine search

Pre- and post-processing hacks, eg. query design

Segmentation, sharding, concurrency and parallelism, and other clever distribution methods—optimizing latency and time and memory complexity

Low-dimensional/cost-optimized retrieval methods for enterprise settings

Identifying, finetuning, and aligning new vector embedding methods

Reranking and boosting from data augmentation and live user interaction emit

Caching mechanisms for high-variance natural language queries

We are a public benefit corporation headquartered in Bangalore. We operate in rapidly changing legal systems with awareness of the stakes at hand. Our intention is to influence beneficence and alignment into the technological systems that are augmenting and replacing human institutions. Our team spans diverse identity and training, from physics and mathematics to law and public policy. We are small, fast-moving, horizontally flat, and built on collaboration between lawyers, academics, and engineers. We ship fast, and every line of code our team writes has a >0.9 expectation of making it to production.

One or more of these might describe you—

Deep familiarity with DL methods and up-to-date with the latest on Arxiv

Proficiency with linear algebra and programming tensor math

Familiarity with complex systems and optimization

Strong engineering skills, ideally with Python proficiency

Excited by large-N data, access to compute, and problems that are emergent with scale

Crave ownership over problem statements

Hungry for impact and to write code that is actually shipped

Prioritize career advancement (eg Chief Scientist title) and growing with a high-velocity company more than immediate compensation

Compensation is negotiable based on levels and mutual excitement.

This role will start ASAP and requires in-person presence for at least 4 days of the week at our Bangalore HQ. Full remote is negotiable for superlative candidates who are not India-based.

Come as you are: we are a diverse team constituted by members of different backgrounds in nationality, religion, caste, gender, and sexual orientation. We sincerely and wholeheartedly welcome diverse individuals.