This website uses cookies to ensure you get the best experience.

Pocketlaw and our selected partners use cookies and similar technologies (together “cookies”) that are necessary to present this website, and to ensure you get the best experience of it. If you consent to it, we will also use cookies for analytics purposes.

See our Cookie Policy to read more about the cookies we set.

You can withdraw and manage your consent at any time, by clicking “Manage cookies” at the bottom of each website page.

Select which cookies you accept

On this site, we always set cookies that are strictly necessary, meaning they are necessary for the site to function properly.

If you consent to it, we will also set other types of cookies. You can provide or withdraw your consent to the different types of cookies using the toggles below. You can change or withdraw your consent at any time, by clicking the link “Manage Cookies”, which is always available at the bottom of the site.

To learn more about what the different types of cookies do, how your data is used when they are set etc, see our Cookie Policy.

These cookies are necessary to make the site work properly, and are always set when you visit the site.

Vendors Teamtailor

These cookies collect information to help us understand how the site is being used.

Vendors Teamtailor
Pocketlaw
Tech · Stockholm · Hybrid

Senior AI Engineer

If you love solving complex problems, enjoy working in a fast-moving environment, and want to shape the future of AI in legal tech, we’d love to hear from you!

At Pocketlaw, we’re revolutionizing legal workflows using AI to make legal work faster, smarter, and more accessible. As a Senior AI Engineer, you will play a crucial role in integrating cutting-edge AI into our products, ensuring they are intelligent, scalable, and efficient. You’ll collaborate closely with our product and engineering teams to create AI-driven features that provide real value to our customers.

If you love solving complex problems, enjoy working in a fast-moving environment, and want to shape the future of AI in legal tech, we’d love to hear from you!

What You’ll Do

  • Design, build, and optimize AI-powered features using Large Language Models (LLMs) like GPT, Claude, Gemini, and Llama.
  • Work closely with product and engineering teams to integrate AI seamlessly into our applications.
  • Improve cost and performance efficiency through advanced techniques like vector databases and retrieval-augmented generation (RAG).
  • Lead technical decisions and contribute to the overall AI and machine learning strategy.
  • Stay up to date with the latest advancements in LLMs and apply best practices to our products.
  • Conduct rigorous testing, experimentation, and evaluation to ensure high-quality AI implementations.

What We’re Looking For

  • Hands-on experience with LLMs (GPT, Claude, Gemini, Llama, etc.)
  • Strong understanding of NLP techniques and text representation methods.
  • Expertise in ML frameworks and tooling (PyTorch, TensorFlow, LangChain, OpenAI APIs).
  • Experience with backend technologies (NodeJS, TypeScript, PostgreSQL).
  • Ability to optimize LLM-powered features for cost and performance.
  • Strong problem-solving skills and ability to turn research into scalable solutions
  • A positive, solution-oriented mindset and passion for innovation.

Our Tech Stack

  • Frontend: React, TypeScript
  • Backend: NodeJS, TypeScript, PostgreSQL
  • AI & ML: PyTorch, TensorFlow, LangChain, OpenAI APIs
  • Infrastructure: GraphQL, Event-driven microservices, Docker, Kubernetes, AWS

Why Join Us?

  • Work on cutting-edge AI-powered products that transform the legal industry.
  • Be part of a collaborative, high-impact team where your work truly matters.
  • Opportunities for growth, leadership, and innovation in a fast-scaling company.

Interested? Apply now or reach out to learn more!

Team
Tech
Locations
Stockholm
Remote status
Hybrid

Job openings

More jobs

Stockholm

The people on this mission

Realising this mission and forever changing how companies solve legal needs requires brilliant people, driven by an eagerness to challenge the status quo and push for change. Luckily, great people are the foundation of our business and we are able to draw on a leadership team which has developed deep and diverse experience working at companies such as Slack, Google, Spotify, Klarna, Mannheimer Swartling and Hogan Lovells.

About Pocketlaw

Pocketlaw helps companies solve their everyday legal needs - empowering every team in the organisation to create contracts and manage legal with efficiency, ease and confidence. Pocketlaw is transforming legal for businesses; turning it from a business blocker into a business enabler - with the end goal of giving every company in the world legal peace of mind.

Founded in 2018
Tech · Stockholm · Hybrid

Senior AI Engineer

If you love solving complex problems, enjoy working in a fast-moving environment, and want to shape the future of AI in legal tech, we’d love to hear from you!

Loading application form

Already working at Pocketlaw ?

Let’s recruit together and find your next colleague.

@pocketlaw.com