Senior AI/ML Software Engineer, Remote

November 5

🏄 California – Remote

Although this job is listed in California, this company may be open to hiring remote in other states.

💵 $150k - $256k / year

⏰ Full Time

🟠 Senior

🧑‍💻 Full-stack Engineer

🦅 H1B Visa Sponsor

Apply Now

Receive Emails with Similar Jobs

Rocket Lawyer

WebsiteLinkedInAll Job Openings

SaaS • Enterprise • B2C

Rocket Lawyer is a legal tech company that offers online legal help for businesses and individuals at an affordable price. They provide personalized legal documents, document review services, e-signatures, legal advice, and assistance with business registration and filings. While Rocket Lawyer is not a law firm, it has been authorized to provide legal services in certain jurisdictions through Rocket Legal Professional Services, Inc. , a non-lawyer-owned entity. With a focus on accessibility and affordability, Rocket Lawyer has assisted over 20 million clients worldwide by streamlining legal processes and providing professional services through its online platform.

201 - 500 employees

Founded 2008

☁️ SaaS

🏢 Enterprise

👥 B2C

📋 Description

• Rapidly prototype product features to explore new AI capabilities, while writing clean, understandable code • Deliver generative AI features to production • Mentor junior AI developers • Create, monitor and troubleshoot backend API services to allow Rocket Lawyer product frontends to interact with ML models • Work closely with multiple product teams and cross-functional teammates to quickly integrate AI into existing and new Rocket Lawyer products • Build, train, fine-tune, deploy and maintain ML models for AI features • Keep up-to-date with industry trends around generative AI and machine learning

🎯 Requirements

• 7+ years of experience writing ML code for launched products • Experience leading a small team of developers • Experience writing production features leveraging LLMs and generative AI • Understanding of the state of the art in generative AI, including RAG, LoRA, RLHF and others • Experience creating data sets from proprietary anonymized data, and using them to train and fine-tune ML models • Strong programming skills in Python • Experience with Pydantic AI • Experience with vector databases • Strong programming skills in NodeJS, Java, Kotlin, or similar languages • Experience or familiarity with AI model evaluation techniques • Interest in building user-facing products • Understanding of microservices architectures and distributed systems • Experience with building data pipelines using technologies such as Snowflake or Hive • Experience with machine learning frameworks like TensorFlow, PyTorch and Scikit-Learn • Strong programming skills in SQL

🏖️ Benefits

• Comprehensive health plans (including Medical, Dental and Vision insurance for full-time employees) • Unlimited PTO • Competitive salary packages • Life insurance • Disability benefits • Supplemental Optional Life Insurance Benefits • FSA Options Optional • HSA with Company Match • 401k program with Company Match • Fertility Assistance and Planning options • Wellhub & ClassPass fitness platforms • Comprehensive Pet Insurance options • Financial Wellbeing & Student Loan Program access • Access to additional Mental Health & Wellbeing resources • Pre-tax Commuter/Transit Benefits • Free Rocket Lawyer account with online access to an extensive legal documents library and brilliant licensed attorneys at discounted rates