About the job
Join our team as a Senior Machine Learning Engineer and help build the infrastructure powering large-scale, multimodal AI training and inference. Collaborate with top researchers, develop distributed systems, and drive innovation in data engineering. Shape the future of AI at scale with cutting-edge tools and a dynamic, supportive environment.
Responsibilities
Build distributed data loaders to support large-scale training workflows
Develop data pipelines for ingesting, transforming, and preparing multimodal datasets
Design batch inference systems for high-volume data processing across GPU environments
Improve system performance, scalability, and reliability using distributed computing tools (e.g., Ray, Spark, DuckDB)
Implement search and retrieval systems using vector databases and embedding-based approaches
Develop and maintain CI/CD workflows, including testing, deployment, and containerization
Partner with researchers and engineers to turn model requirements into scalable systems
Create reusable tools, libraries, and documentation to support teams across the organization
Monitor and improve system health, including throughput, latency, and resource utilization
Support a collaborative team environment through code reviews and knowledge sharing
Qualifications
Minimum
8+ years of experience building and operating distributed systems or ML infrastructure in production
Experience working with large-scale data pipelines or inference systems
Strong programming skills in Python and a foundation in software engineering principles
Experience with ML frameworks such as PyTorch or TensorFlow
Familiarity with distributed computing tools (e.g., Ray, Spark, Dask, or similar)
Experience working with cloud platforms such as AWS or Azure
Understanding of MLOps practices, including CI/CD and deployment workflows
Ability to communicate clearly and collaborate with cross-functional teams
Preferred
Experience working with multimodal data (images, video, text)
Familiarity with vector databases or semantic search systems