Member Of Technical Staff (AI)

TL;DR

Member Of Technical Staff (AI): Building scalable infrastructure for AI model training and inference with an accent on GPU optimization. Focus on defining technical direction and building core systems that power the GPU optimization platform.

Location: San Francisco, CA, US

Salary: $6K - $10K / monthly

Company

AI that makes AI fast.

What you will do

• Build scalable infrastructure for AI model training and inference
• Lead technical decisions and architecture choices

Requirements

• GPU Fundamentals: Deep understanding of GPU architectures, CUDA programming, and parallel computing patterns.
• Deep Learning Frameworks: Proficiency in PyTorch, TensorFlow, or JAX, particularly for GPU-accelerated workloads.
• LLM/AI Knowledge: Strong grounding in large language models (training, fine-tuning, prompting, evaluation).
• Systems Engineering: Proficiency in C++, Python, and possibly Rust/Go for building tooling around CUDA.

Culture & Benefits

• Connect directly with founders of the best YC-funded startups.