Ai jobs in Singapore

Job Description

Alexi is transforming the legal industry through artificial intelligence, backed by $15M in Series A funding from leading investors Drive Capital and Draper Associates. As a fast-growing legal AI platform in Canada, we're revolutionizing how lawyers and litigation professionals work to achieve the best outcomes for their clients by automating complex legal tasks and workflows.

Our AI-powered platform serves over 1,000 lawyers across leading law firms in the US and Canada. We've achieved rapid growth, doubling our customer base in 2024 alone, while increasing the engagement on the platform by 22x.

At Alexi, we believe that transforming the legal industry requires more than just innovative technology—it demands a culture of customer obsession, ownership, and continuous learning. We’re on a mission to transform legal work through AI.

About the Role

As a Staff Engineer, you will be a key technical leader, driving architectural decisions, mentoring, and ensuring the scalability, reliability, and performance of our Python-based AI-driven applications. You will collaborate across teams to deliver high-impact solutions while maintaining best practices in software engineering.

What You'll Do

• Own delivery of large scale initiatives that align with company objectives.
• Drive architectural decisions, ensuring code quality, maintainability, and adherence to best practices.
• Collaborate with cross-functional teams, including Platform, Data Engineering, Applied AI and Product, to align technical strategies with business goals.
• Optimize and enhance system performance, reliability, and security.
• Create and own SLIs, SLOs and their monitoring that aligns with business goals.
• Advocate for SRE principles and foster a culture of reliability within the organization.
• Leverage cloud infrastructure (AWS, GCP, Azure) and GPU/TPU acceleration for efficient model inference.
• Optimize LLM inference and training pipelines, focusing on speed, cost efficiency, and scalability.
• Develop and own a comprehensive test strategy aligned with project requirements and business goals that covers unit, integration, e2e, performance, and security testing.
• Develop and own bug tracking and resolution processes.
• Build out infrastructure using IaC.
• Advocate for best practices in CI/CD, testing, and automation.
• Stay up to date with emerging technologies, frameworks, and methodologies in the Python ecosystem.
• Lead incident response and troubleshooting for critical issues in production.
• Design and create security best practices across the organization that align with security best practices.
• Design and implement effective local development workflow across multiple projects.

Requirements

• 8+ years of software engineering experience with Python.
• Strong expertise in designing and building scalable distributed systems, microservices, and APIs.
• Deep understanding of asynchronous programming, concurrency, and performance optimization.
• Experience with cloud platforms (AWS, GCP, or Azure) and containerization (Docker, Kubernetes).
• Proficiency in database design and optimization (SQL & NoSQL).
• Strong problem-solving skills and the ability to lead high-impact technical initiatives.
• Experience with CI/CD pipelines, Infrastructure as Code (Terraform, CloudFormation), and DevOps practices.
• Strong experience with observability tools (logging, tracing, monitoring) and service level indicators
• Strong problem-solving skills and a track record of technical leadership.
• Excellent communication skills with the ability to collaborate effectively across teams.
• Knowledge of event-driven architectures and messaging systems
• Experience integrating LLMs with APIs, chatbots, and real-time applications.
• Experience in vector databases (e.g., FAISS, Pinecone, Weaviate) and retrieval-based AI.
• Experience with static and dynamic application security testing (SAST/DAST) tools to identify security flaws.
• Experience building secure API development with authentication (OAuth, JWT, mTLS) and authorization (RBAC, ABAC).
• Experience with Cloud Security Posture Management (CSPM) and Infrastructure as Code (IaC) security scanning (Terraform, CloudFormation).
• Experience building internationalized and multi regional applications. ( Nice to have
• Experience with graph based LLM frameworks and related technologies (Nice to have).

Alexi is an equal-opportunity employer. We celebrate and are committed to creating an inclusive environment for all employees.

If you have the above experience and skills APPLY NOW and we will be in touch with all suitable candidates.