About
Own the training pipeline for large-scale LLM fine-tuning and post-training workflows
Configure, launch, monitor, and debug multi-node distributed training jobs using FSDP, DeepSpeed, or custom wrappers
Contribute to upstream and internal forks of training frameworks like TorchTune, TRL, and Hugging Face Transformers
Tune training parameters, memory footprints, and sharding strategies for optimal throughput
Work closely with infra and systems teams to maintain the health and utilization o...
Updates & Highlights
Recent Openings
View all 2 jobs arrow_forwardMachine Learning Engineer - Training & Infrastructure
location_on San Francisco, CA
All Jobs
Machine Learning Engineer - Training & Infrastructure
location_on San Francisco, CA
Culture & Mission
Benefits & Perks
Equity/Stock Options
Ownership in the company
Health Insurance
Medical, dental & vision
Remote Work
Work from anywhere flexibility
Learning Budget
Professional development
Generous PTO
Paid time off
401(k) Match
Retirement savings
Tech Stack
Core Values
Innovation
Pushing boundaries in AI/ML technology
Collaboration
Working together to achieve more
Excellence
Striving for the highest quality
Impact
Making a difference in the world
Office Locations
Team
Are you on the team at P-1 AI?
Claim this company page to verify ownership, edit details, and respond to applicants.
Claim this company arrow_forwardNews & Press
4 recentP-1 AI: $23 Million Raised For Building Engineering AGI For Physical Systems
P-1 AI: $23 Million Raised For Building Engineering AGI For Physical Systems
Pulse 2.0 · about 1 year ago
How is P-1 AI Building Engineering AGI for the Future?
How is P-1 AI Building Engineering AGI for the Future?
Analytics India Magazine · about 1 year ago
The adolescence of P-1 AI
The adolescence of P-1 AI
Engineering.com · about 1 year ago
P-1 AI Comes Out of Stealth, Aims to Build Engineering AGI for Physical Systems
P-1 AI Comes Out of Stealth, Aims to Build Engineering AGI for Physical Systems
Business Wire · about 1 year ago