Machine Learning Scientist
Advances small, high-performance language models for retrieval, application, and code generation. Requires 2+ years ML research/production experience, Python/PyTorch/JAX fluency, optimization expertise, and advanced degree.
About the job
Requirements
- Strong background in machine learning, deep learning, or related fields
- 2+ years of experience working on ML research or production systems
- Fluency in Python and frameworks like PyTorch or JAX
- Experience with training and optimizing large or efficient models
- Strong understanding of applied optimization, distributed training, or model evaluation
- Familiarity with code models, retrieval systems, or language modeling a plus
- Advanced degree (MS or PhD) in a quantitative field, or equivalent industry experience
- Willingness to work in-person from our SF office in FiDi
Skills
Machine Learning, Deep Learning, PyTorch, JAX, Python, Distributed Training, Model Optimization, Language Modeling, Retrieval Systems, Code Models
Similar jobs
ML Engineering jobs
DataVisor
Mountain View, CA
Builds high-scale data pipelines, distributed systems, and AI agent workflows using LLMs for fraud intelligence platform. Requires 2+ years software engineering, Python proficiency, big data tools, AWS/K8s, and ML foundations.
Coinbase
San Francisco, CA
Machine learning intern pursuing a Ph.D. who will build and deploy production-scale models and pipelines, conduct an end-to-end research project, and collaborate with engineering and product teams on blockchain and cryptocurrency applications.
Fireworks AI
San Mateo, CABuild, deploy, and optimize AI applications and machine learning models for customer use cases while contributing to an internal ML platform. This new graduate role requires a technical master’s degree, hands-on ML or LLM experience, and strong customer communication skills.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI product development, collaborating with scientific, engineering, and product teams. Requires a master's degree with 2–4 years of experience or a PhD with 0–2 years, plus strong Python and ML development skills.
Earnin
Mountain View, CA
Machine learning engineer who trains, evaluates, and productionizes models and LLM-powered applications for financial products. Requires 2+ years of ML systems experience, strong Python and PyTorch skills, production data pipelines, model evaluation, and API development.