Staff Software Engineer (LLMs)

AI71 is a fast-growing UAE-based startup focused on bringing the benefits of Generative AI to the world. As a member of the team, you will contribute to the development of industry-leading AI software, leveraging cutting-edge research breakthroughs. Your role will involve designing, implementing, launching, and maintaining code within a dynamic and innovative environment, combining both platform and project responsibilities.

Role Responsibilities:

• Design and implement advanced language model-powered solutions to enhance task efficiency and productivity.
• Enhance API and interface performance, ensuring scalability and user-friendly design.
• Customize user experience for accessibility and smooth integration into daily workflows.
• Collaborate with research teams to refine interfaces based on user feedback and real-world usage data.
• Develop and deploy systems to monitor model performance and alignment with user requirements.
• Pioneer innovative human-AI interaction paradigms.

Required Qualifications:

• Bachelor's degree or equivalent practical experience.
• 6+ years of software development experience in one or more programming languages, or 1 year with an advanced degree.
• 4 years of experience with AI, data structures, or algorithms (mandatory).
• Hands-on experience in fine-tuning and deploying generative LLMs using TensorFlow, XLA/MLIR, JAX, and PyTorch frameworks (mandatory).
• Experience developing RAG models integrated with vector databases to enhance conversational AI products (mandatory).
• Familiarity with autoregressive sequence models (mandatory).
• Excellent in-depth understanding of NLP, LLM, MLLM, Generative AI, and RAG workflows (mandatory).

Work Conditions: Hybrid, Full-time

Location: Abu Dhabi, United Arab Emirates

Language Requirements: Not specified. Please only apply if you meet the mandatory experience qualifications.