Skip to content
PulsePulseSan Francisco, CA

Machine Learning Engineer

Develops and fine-tunes specialized vision and language models for document understanding, including OCR, layout, tables, and charts. Requires 3+ years ML experience with PyTorch/JAX and strong engineering focus; onsite in San Francisco.

160k – 220k/yr
On-site3+ YOEML Engineering

About the role

Responsibilities

  • Train and fine tune OCR, layout, table, and vision-language models
  • Build evaluation, data curation, and active learning pipelines
  • Optimize inference, batching, and quantization on GPU
  • Productionize models with clear SLAs and rollback plans
  • Write internal notes that inform model and product roadmaps

Requirements

  • 3+ years in applied ML or research, or strong open source record
  • PyTorch or JAX, and modern vision or multimodal architectures
  • Solid engineering discipline and metrics focus

Nice to Have

  • Triton Inference Server, TensorRT, ONNX, distributed training

Compensation and Benefits

  • Competitive base salary plus equity, performance-based bonus
  • Relocation assistance for Bay Area moves
  • Daily meal stipend
  • Medical, vision, and dental coverage

Skills

PyTorchJAXOcrVision-Language ModelsMultimodal ArchitecturesTriton Inference ServerTensorRTOnnxDistributed TrainingActive LearningGpu Optimization

Similar roles

ML Engineering jobs
Pindrop

Research Scientist II

PindropUnited States

Research Scientist II building and improving fraud risk models and scam detection systems using audio, behavioral, and metadata signals. Requires an advanced degree and 3+ years of applied ML experience with Python and modern ML frameworks.

160k – 185k/yr
Remote3+ YOEML Engineering
Snowflake

AI Engineer - Database Engineering

SnowflakeMenlo Park, CA

As an AI Engineer, you will own the full AI engineering lifecycle, from design to optimization, for Snowflake Database Engineering products. You will build agentic workflows, coding harnesses, and evaluation pipelines, working with a high-powered engineering team.

160k – 230k/yr
On-site5+ YOEML Engineering
Snowflake

Software Engineer, Cortex AI Infrastructure

SnowflakeMenlo Park, CA

Build and scale backend infrastructure powering agentic AI products including orchestration engines, RAG systems, evals infrastructure, and production AI workflows. Requires 4+ years distributed systems experience and deep Python plus Go/Java proficiency.

160k – 225k/yr
On-site4+ YOEML Engineering
Mach9

ML Infrastructure Engineer

Mach9San Francisco, CA

Builds and maintains ML infrastructure for training pipelines handling massive 3D data and real-time inference serving integrated with CAD software. Requires 3+ years experience with Python, PyTorch, ML orchestration tools, data versioning, and inference optimization.

160k – 200k/yr
On-site3+ YOEML Engineering
Deepgram

ML Ops Infrastructure Engineer

DeepgramCalifornia

Build and maintain ML infrastructure pipelines to deploy research models to production at scale, including CI/CD, A/B testing, monitoring, and optimization for low-latency voice AI serving. Requires 4+ years MLOps experience with Python, Docker, Kubernetes.

160k – 220k/yr
Remote4+ YOEML Engineering