Jobs in Klang, Malaysia

Flank is the leading Agentic AI platform for in-house legal teams-built for a future where autonomous AI colleagues handle entire workflows.

Companies like TravelPerk, QA, and Mural use Flank to automate contract negotiation, email triage, and security reviews. What used to take hours of manual work now happens automatically.

We're building in an emerging market that's reshaping how businesses operate. Instead of incremental software improvements, we're designing entirely new ways for AI and humans to collaborate.

If you're excited about defining what's possible when AI becomes a true colleague, you'll find a home here.

We're seeking an experienced AI Engineer to drive our applied AI initiatives.

We're a fast-moving start-up where you'll work closely with product, design, and commercial teams to ship real AI into production - not just research experiments.

You will be tasked with leveraging emerging technologies and a team of legal experts to improve the intelligence and autonomy of our agents.

This might start with building an evals framework in collaboration with our team of lawyers, it might spawn a hypervisor that governs every step of our agentic system and that "knows what good looks like", and it might end with us building the first alignment program for legal AI systems.

What you'll be doing:

• Drive Applied Research: Drive applied research into LLMs and agentic systems, rapidly prototyping, validating, and shipping production-ready capabilities.
• Stay Ahead of the Curve: Stay hands-on with the latest advances in foundation models, LLM-as-a-judge, agentic frameworks & tools (e.g. MCP), and move fast to bring them into the product.

• Demonstrated Experience: A strong track record of leading and delivering cutting-edge AI initiatives that have driven meaningful product and business impact. Hands-on experience building and deploying ML/AI products is essential.
• Technical Expertise:
  
  • Strong programming skills in Python (and familiarity with TypeScript/Go a plus).
  • Experience with deep learning frameworks (PyTorch, TensorFlow, Hugging Face).
  • Hands-on experience deploying ML/LLM models in production environments.
  • Solid understanding of data pipelines, APIs, and distributed systems.
  • Knowledge of prompt engineering, RAG, fine-tuning, and embeddings for LLMs.
  • Experience with cloud platforms (AWS, GCP, or Azure) and containerization (Docker, Kubernetes
• Thought Leadership: Ph.D. or Master's in Computer Science, Machine Learning, or a related field from a leading academic institution.
• Cross-Functional Collaboration: Proven ability to build strong partnerships across product, engineering, and research teams to drive innovation.

• Innovative Work: Be part of a team making bold bets to create a new paradigm in AI-enhanced legal services.
• High Standards Culture: Join a team that demands excellence and accountability through direct and candid communication.
• Empowered Environment: Operate in a high-agency culture where you can lead yourself and make a real impact.
• Global Team: Remote-first, collaborate with a diverse, talented team across the globe.
• Competitive, Benchmarked Salary