Member of Technical Staff – GPU Performance Engineer

🕒 Março 27

Candidatar-se

Encontrar Vagas Remotas Similares

Configurar Alerta de Email para Novas Vagas

📊 Verifique sua pontuação de currículo para esta vaga

Melhore suas chances de conseguir uma entrevista verificando sua pontuação de currículo antes de se candidatar.

Enviar Currículo

Reka AI

SiteLinkedInTodas as Vagas

1 - 10 funcionários

Fundada em 2022

🤖 Inteligência Artificial

🏢 Corporativo

☁️ SaaS

💰 $58.000.000 Series A em 2023-06

Artificial Intelligence • Enterprise • SaaS

Reka AI é uma empresa especializada no desenvolvimento de modelos de IA multimodal de ponta. Esses modelos são projetados para capacitar agentes de IA com a capacidade de ver, ouvir e falar, facilitando uma gama de aplicações que vão desde texto e código até imagens, vídeos e dados de áudio. Os modelos proprietários da Reka, incluindo Reka Core, Flash, Edge e Spark, são criados do zero com arquiteturas inovadoras que permitem a implantação em dispositivos, no local ou na nuvem. Conhecida por avançar no campo da pesquisa em IA, a Reka AI colabora com parceiros tecnológicos globais para expandir o acesso às suas avançadas capacidades de IA multimodal.

Descrição

• Design and implement improvements to training infrastructure • Directly contribute to technical decisions optimizing performance of models • Work on post-training processes, including reinforcement learning and fine-tuning • Improve the efficiency and scalability of model serving infrastructure

🎯 Requisitos

• Strong engineering skills with fluency in Python and PyTorch (or other frameworks) • Proven experience implementing and training large deep learning models • Experience writing and debugging low-level GPU code (CUDA, C++) • Experience scaling up GPU jobs using large-scale compute clusters (e.g., Slurm or Kubernetes) • Demonstrated ability to analyze and optimize the performance of GPU-accelerated workloads, including profiling, identifying bottlenecks, and implementing performance tuning techniques

🏖️ Benefícios

• Five weeks of paid leave to recharge • Comprehensive healthcare benefits (including vision and dental) • Additional perks that support your well-being • Visa assistance, including H1B and OPT transfers, for US employees