ML Engineer (ONSITE IN SF)

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

Dear applicants, please keep in mind that applications without provided salary expectations and active LN profile will not be considered. Hope for your understanding.
Location: San Francisco, CA (In-person)
Employment Type: Full-Time
Equity: 0.5% – 1%
Visa: Not available
Experience: 1+ years (exceptional new grads welcome)

We are hiring ML Engineers to implement research ideas reliably and operate full training pipelines end-to-end. This is not a research-only role. This is research-engineering at scale. A seed-stage research-driven ML company focused on mechanistic understanding of model architectures and optimizers.
The team studies:
  • Optimizer–architecture co-design
  • Orthogonalized optimizers and manifold-based training
  • Sparse attention mechanics
  • Data-efficient reasoning models
  • Learning dynamics in data-sparse regimes
The environment blends academic rigor with industrial compute and speed. The team is deliberately long-term oriented and avoids premature commercialization pressure.
You will:
  • Translate research papers into working PyTorch/JAX implementations
  • Run distributed transformer training
  • Debug divergence and instability
  • Optimize throughput
  • Build full pipelines (data → training → evaluation)
  • Reason about learning dynamics and architecture tradeoffs
  • The bar is slope and research intuition, not years.

What You’ll Own
  • Reliable implementation of novel architectures
  • Distributed transformer training at scale
  • Training stability and performance debugging
  • Evaluation frameworks
  • Optimization reasoning alongside researchers

Must-Have Requirements
  • Strong PyTorch or JAX proficiency
  • Hands-on transformer training experience
  • Experience with distributed training setups
  • Debugging divergence and instability
  • Ability to read and implement research papers
  • Research intuition around optimization and learning dynamics
  • High growth slope

Nice to Have
  • Megatron-LM, DeepSpeed, xformers
  • End-to-end pipeline ownership
  • Research-engineering team experience
  • Mathematical depth (optimization, information theory, etc.)
  • Competitive programming / theory-heavy background

PulseRise Technologies

PulseRise Technologies — The Pulse of Global IT Talent At PulseRise Technologies, we don’t just build tech teams — we breathe life into innovation. We are an international IT outstaffing company based in Cyprus, helping ambitious organizations across Europe, LATAM, and USA connect with exceptional technology professionals. We are the pulse behind global success stories — keeping companies in motion, projects in sync, and ideas alive. Because every thriving company needs one thing: a heartbeat that never stops. Headquartered in Cyprus, PulseRise Technologies operates across Europe and Latin America, bridging businesses and talent through a multilingual, multicultural team. We understand local markets, respect global standards, and align perfectly with your working culture. From emerging startups to enterprise giants, PulseRise Technologies is the silent rhythm that keeps your delivery on time, your teams inspired, and your goals alive. Because when your company finds its pulse — everything flows. Our essence inspired by the word “pulse” — the rhythm, heartbeat, and energy that powers every connection. Just as a pulse gives life to the body, PulseRise Technologies keep your business alive, vibrant, and constantly moving forward. Our Services: 🔹 Software & Web Development – Scalable, secure, and tailored to your goals. 🔹 Cloud & DevOps Engineering – Fast, reliable, and automated for performance. 🔹 Data Analytics & BI – Turning numbers into decisions that drive growth. 🔹 QA & Automation Testing – Precision that ensures perfection. 🔹 UI/UX & Product Design – Interfaces that connect logic with emotion. 🔹 L1–L3 Technical Support – Always-on reliability that builds trust with your users. We build high-performing remote tech teams and deliver end-to-end solutions that keep your business agile, competitive, and cost-efficient.
Runway Icon
Boost Your Interview Chances

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