Pocketlaw

Senior AI Engineer

Stockholm, Stockholm County, SE

2 months ago
Save Job

Summary

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!

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.

How strong is your resume?

Upload your resume and get feedback from our expert to help land this job