3 months ago

AI Systems, Training

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 6 hours agoStatus: Live

Job Summary

Senior ML hardware–software role focused on building and optimizing a next-generation ML model training platform. You’ll co-design training systems, develop multi-node distributed training, implement elastic data streaming and checkpointing, and optimize kernels (CUDA/Triton) while coordinating with theory and hardware teams to push physics-based compute boundaries in a Palo Alto/Remote hybrid setting.

Location: Palo Alto

Workplace: Hybrid

Employment Type: Full time

Job Function: Education & Training

Key Responsibilities

•Build and maintain highly optimized, model-specific training stacks for state-of-the-art generative vision, language, and world models.
•Design and scale multi-node distributed training systems with elastic sharding and robust data streaming pipelines; implement and robust model checkpointing and recovery mechanisms.
•Develop and optimize kernels using low-level programming models like CUDA and Triton; design benchmarking suites to track MFU, memory bandwidth, and convergence stability.
•Act as a translator between theorists and infrastructure/hardware teams to convert model requirements into concrete specifications.

Pay and Benefits

Perks:Health Insurance401kPaid Leave

Key Requirements

•MS/PhD or equivalent research/project experience in AI/ML, CS, Physics, EE, or Applied Math.
•Experience mapping state-of-the-art AI model architectures (e.g., transformers, Mixture of Experts, diffusion models) to system performance implications and understanding of cluster partitioning, communication primitives, and parallelism strategies.
•Proven track record of implementing, debugging, and maintaining production-grade training frameworks (Megatron-LM, DeepSpeed, Ray, PyTorch Lightning).

Experience:AIMachine LearningDeep Learning

Education:PhD / Doctorate

Skills:CommunicationCollaborationProblem-solving

Languages:English

Tech Stack:CUDATritonMegatron-LMDeepSpeedRayPyTorch Lightning

Company Brief

Unconventional AI

Builds customizable AI copilots and fine-tuned large language model solutions to help teams automate workflows, surface actionable insights, and integrate generative AI into business processes across products and operations.

Industry: AI & Machine Learning

Company Size: Small (11 to 50 employees)

Growth: Early Stage Startup

Funding: Seed

Website

Loading job details...

Preparing the role view and application actions.

Unconventional AI

3 months ago

AI Systems, Training

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 6 hours agoStatus: Live

Job Summary

Location: Palo Alto

Workplace: Hybrid

Employment Type: Full time

Job Function: Education & Training

Key Responsibilities

•Build and maintain highly optimized, model-specific training stacks for state-of-the-art generative vision, language, and world models.
•Design and scale multi-node distributed training systems with elastic sharding and robust data streaming pipelines; implement and robust model checkpointing and recovery mechanisms.
•Develop and optimize kernels using low-level programming models like CUDA and Triton; design benchmarking suites to track MFU, memory bandwidth, and convergence stability.
•Act as a translator between theorists and infrastructure/hardware teams to convert model requirements into concrete specifications.

Pay and Benefits

Perks:Health Insurance401kPaid Leave

Key Requirements

•MS/PhD or equivalent research/project experience in AI/ML, CS, Physics, EE, or Applied Math.
•Experience mapping state-of-the-art AI model architectures (e.g., transformers, Mixture of Experts, diffusion models) to system performance implications and understanding of cluster partitioning, communication primitives, and parallelism strategies.
•Proven track record of implementing, debugging, and maintaining production-grade training frameworks (Megatron-LM, DeepSpeed, Ray, PyTorch Lightning).

Experience:AIMachine LearningDeep Learning

Education:PhD / Doctorate

Skills:CommunicationCollaborationProblem-solving

Languages:English

Tech Stack:CUDATritonMegatron-LMDeepSpeedRayPyTorch Lightning

Company Brief

Unconventional AI

Industry: AI & Machine Learning

Company Size: Small (11 to 50 employees)

Growth: Early Stage Startup

Funding: Seed

Website