PulseRise Technologies
PulseRise Technologies
Posted about 2 months ago
Full Time
San Francisco, California
In Person
Smart Summary
Responsibilities
You will translate research papers into functional PyTorch or JAX implementations and operate full-scale distributed transformer training pipelines. Additionally, you will be responsible for debugging training instability and optimizing throughput for research-driven model architectures.
Qualifications
You have strong PyTorch or JAX proficiency and hands-on experience with transformer training, including distributed setups. You are adept at implementing research papers, debugging training issues, and building end-to-end ML pipelines with a focus on optimization and learning dynamics.
Must Have Skills for ATS
PyTorch
JAX
transformer training
distributed training
Megatron-LM
DeepSpeed
xformers
Job Description
PulseRise Technologies
With Runway
See Your Fit for This Role
1-5 min
Your Score
?
Top Applicants
90%
Your Job Search Advantage
Key Gaps & Next Steps:
Address these in your resume & Interview
Top Strengths For This Role
Highlight these in your cover letter & interview
Your Interview Guide
A Personalized Interview Strategy
Freshest Opportunities
Never Miss a Good Fit
Get notified when jobs mach your criteria