About this role
Research Engineer, Infrastructure, Kernels at Thinking Machines Lab. Location: San Francisco, California, United States. Role: Design kernels, Optimize kernels, Collaborate researchers Requirements: Design, optimize, and maintain ML compute foundations; develop high-performance kernels; collaborate with researchers; strong CUDA/Triton experience. Category: Research and Development (R&D) Seniority: Mid Level Tools: CUDA, CuTe, Triton, PyTorch, JAX Commitment: Full Time Workplace: Onsite Languages: English