About this opportunity
kadence lists this Staff Machine Learning Engineer opportunity in san francisco, California. Review the employer’s description below for duties, qualifications and application requirements.
Job description
Base Pay Range
$180,000.00/yr - $250,000.00/yr
Direct message the job poster from kadence
Senior Consultant | Kadence | AI + ML Talent Solutions
AI Infrastructure Company | $40M Series A
About the Company
We are partnered with a developer-focused, open-source data infrastructure company that is led by a world-class team of open-source builders (including contributors to foundational data and storage technologies). Their platform powers cutting‑edge multimodal AI applications, from large‑scale vector search to advanced retrieval workflows and high‑performance data pipelines.
They’re building the next‑generation foundation for AI teams working with large, complex, multimodal datasets.
About the Role
We’re looking for a Machine Learning Engineer with hands‑on experience in model development (training, fine‑tuning, feature engineering) and a strong background in scalable data/ML systems. In this role, you’ll build core components of our multimodal data platform, shape the developer experience for AI teams, and collaborate with forward‑thinking customers and partners.
Responsibilities
Act as an in‑house expert on modern AI engineering, with working experience across frameworks such as PyTorch or JAX.
Drive a best‑in‑class Developer Experience to accelerate productivity for AI/ML practitioners.
Lead the design and development of high‑performance, large‑scale feature engineering infrastructure.
Partner closely with customers, design partners, and the broader open‑source community.
Requirements
3+ years of experience building or deploying ML/DL models in production, or developing infrastructure that supports them.
Strong Python skills and experience with frameworks such as PyTorch or TensorFlow.
Demonstrated ability to own projects end‑to‑end, from design through delivery.
Working knowledge of major cloud platforms (AWS, GCP, or Azure), including managed storage and compute services.
Familiarity with monitoring/logging stacks like Prometheus, Grafana, or ELK/EFK.
Nice to Have
Deep understanding of training architectures (PyTorch/JAX internals, CUDA kernel optimization, TPU environments).
Experience building or managing feature stores (Feast, Tecton) or custom feature registries.
Knowledge of Docker internals, Kubernetes, Slurm, and scheduling/orchestration systems.
Strong Python engineering or Rust background.
Prior experience working directly with customers.
Experience building advanced monitoring and observability systems.
Hands‑on experience with Kubernetes, Terraform, Docker, CI/CD tooling.
Familiarity with large‑scale data/compute ecosystems (Spark, Flink, Delta Lake, Ray, Dataflow, Kafka, Airflow, Kubeflow, etc.).
Why Join
You’ll join a deeply technical, mission-driven team building the infrastructure that will underpin the next wave of AI. You’ll work on open‑source technology, shape core system architecture, and directly influence how top‑tier AI teams interact with multimodal data at scale.
Seniority Level
Mid‑Senior level
Employment Type
Full‑time
Job Function
Engineering and Research
Industries
Technology, Information and Media and Software Development
#J-18808-Ljbffr
Worksite address
san francisco, CA, 94199, US
Who can apply
Review the original listing for work authorization, qualifications and employer requirements.