Senior Machine Learning Engineer

About Nomi.ai: Nomi.ai is building AI companions that form deeply meaningful, humanlike relationships and immersive roleplaying experiences. With over a million users growing at ~10% month-over-month, your work directly impacts millions of lives. Our users tell us we’ve helped them find self-worth, leave unhealthy relationships, try therapy, and even save their lives. Our product is backed by extensive research and positive user testimonials, as seen in our User Spotlight section.

What We Offer:

• Full autonomy to experiment and deploy cutting-edge ML techniques
• A fully remote, async culture emphasizing results over meetings
• International team with visa sponsorship available
• For US employees: 401k (100% match up to 5%), fully covered health insurance, equity

About The Role: We are seeking a Senior Machine Learning Engineer to lead innovation in large language model (LLM) post-training, retrieval augmented generation (RAG), and agentic capabilities, directly shaping how users connect with our AI.

Responsibilities:

• Lead innovation in LLM post-training, RAG, and agentic capabilities, directly shaping how users connect with our AI.
• Design and implement systems that optimize network traffic and scale for global expansion.
• Drive architectural decisions for high-impact projects, ensuring scalability and reliability.
• Co-author long-term technical roadmaps that set the vision for network scalability, performance, and engineering velocity.
• Set the standard for a culture of engineering and operational excellence.
• Build and nurture effective partnerships across engineering, product, and security teams to align on network infrastructure goals.
• Lead and conduct design reviews for critical projects with a focus on system-level tradeoffs and network scalability.
• Drive the design and implementation of “secure by default” network systems in collaboration with security teams.
• Act as a key stakeholder by engaging with customers and internal teams to understand business requirements and deliver innovative solutions.

Requirements:

• Proven experience leading complex engineering efforts focused on network traffic management, network optimization, and cloud orchestration.
• Exceptional collaboration and communication skills, with a demonstrated ability to align stakeholders and lead cross-functional teams.
• A track record of contributing to long-term technical roadmaps and making thoughtful system-level tradeoffs.
• At least 8+ years of coding experience in languages like Go, Java, or similar, with strong expertise in writing concurrent and distributed code.
• Extensive experience designing and building distributed systems, leveraging concurrency primitives, and optimizing network performance.
• Deep expertise in traffic and networking systems, with familiarity across adjacent infrastructure domains.
• Hands-on experience with cloud providers such as AWS, GCP, or Azure, and the ability to optimize cost, performance, and scalability.
• A strong sense of ownership and the ability to balance short-term priorities with long-term strategic goals.

Great to haves:

• Extensive hands on experience with things such as multinode training, rlhf, knowledge distillation, test time compute, rag.
• Up to date with SOTA in LLM post training (would love to hear what research paper you think is or would be most impactful for our roadmap!).
• Genuine passion for our product, finds the idea of engaging with our community on Discord/Reddit to be a pro (it is a very different experience than developing enterprise software!).
• A high internal bar for excellence and relentless drive.

Compensation: The estimated pay range for this role is $150,000 - $250,000. Additionally, this role is eligible to participate in Nomi’s equity plan.