Principal AI Research Engineer, Redis

• Experience in designing and developing machine learning models, especially in natural language processing
• Deep understanding of transformer architecture, LLMs, and embedding models
• Strong proficiency in Python, including experience with ML/NLP libraries (e.g., TensorFlow, PyTorch, Hugging Face Transformers)
• Experience with at least one low-level system language (c, c++, rust)
• Experience with CUDA and high-performance computing
• Self-directed, ambitious, and eager to learn new skills and technologies
• Master’s or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field

Desirable

• You will join our core AI team to research, design, and implement advancements in how companies run Large Language Models (LLMs) to optimize for relevance, latency, and privacy
• You will tackle complex challenges related to model optimization, fine-tuning, and tool development, impacting millions of users worldwide
• Optimize and fine-tune LLMs or embedding models
• Develop and enhance tools and frameworks built around model orchestration
• Research, design, and implement state-of-the-art algorithms to optimize models at scale for latency and efficiency
• Investigate complex NLP problems driven by custom use cases
• Implement hardware-specific optimizations (e.g., GPUs, TPUs, CUDA)