Founding AI/ML Research Engineer

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills — writing robust, production-grade systems Experience with GPU optimization: memory efficiency, quantization, mixed precision Comfortable owning ambiguous, zero-to-one technical problems end-to-end

Build end-to-end training pipelines Design new model architectures or adapt open-source frontier models Fine-tune models using state-of-the-art methods Architect scalable inference systems Build data systems for training data Develop alignment, safety, and guardrail strategies Design evaluation frameworks Own deployment and GPU optimization Shape early product direction and build AI-powered experiences Explore frontier techniques