About this role
Member of Technical Staff - Training at Unconventional, Inc.. Location: Palo Alto or Palo Alto. Role: build training, design distributed, optimize kernels Requirements: MS/PhD or equivalent in AI/ML, CS, physics, EE, or applied math; deep ML software stack experience; mapping AI models to system performance; production-grade training frameworks (Megatron-LM, DeepSpeed, Ray, PyTorch Lightning). Category: Engineering Seniority: Senior Level Tools: CUDA, Triton, Megatron-LM, DeepSpeed, Ray, PyTorch Lightning Commitment: Full Time Workplace: Hybrid Languages: English