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.
About the job
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
PyTorch, JAX, Ocr, Vision-Language Models, Multimodal Architectures, Triton Inference Server, TensorRT, Onnx, Distributed Training, Active Learning, Gpu Optimization
Similar jobs
ML Engineering jobs
Pindrop
United 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.
Office Hours
San Francisco, CABuild reproducible systems for AI model benchmarking, including datasets, evaluation pipelines, containerized environments, scoreboards, and analysis tools. The role requires 4+ years of professional engineering experience, strong Python, dataset rigor, and Docker expertise.
Applied Intuition
Sunnyvale, CA
Develop and deploy ML-first behavior prediction and planning systems for autonomous vehicles, forecasting the motion and interactions of road users. Requires a bachelor's degree, deep learning lifecycle expertise, and at least three years of production software experience with C++ or Python.
LangChain
New York, NYBuild and teach reliable AI agent systems through customer workshops, technical content, guidance, and reference implementations. The role requires strong Python and agent-development experience plus a background delivering customer-facing technical training.
Roboflow
San Francisco, CA
Build reinforcement-learning environments, evaluations, datasets, and scalable infrastructure for frontier AI capabilities. The role suits a high-agency generalist engineer with experience in agents, evaluations, or RL workflows and strong communication skills.