[Remote] Member of Technical Staff (Machine Learning Engineer)

Note: The job is a remote job and is open to candidates in USA. Reka AI is a globally distributed foundation model startup focused on building useful multimodal artificial intelligence. They are seeking a Member of Technical Staff (Machine Learning Engineer) to translate research into production-ready systems, design end-to-end ML models, and collaborate with cross-functional teams to deliver scalable ML solutions. Responsibilities • Translate cutting-edge research into production-ready machine learning systems • Design, build, and deploy end-to-end ML models and pipelines • Develop and optimize models for image and video processing • Own the full ML lifecycle: experimentation, training/fine-tuning, evaluation, and deployment • Rapidly prototype using open-source models and adapt them for product needs • Conduct experiments, analyze results, and iterate to improve performance • Collaborate with researchers and cross-functional teams (product, engineering, design) to deliver ML solutions at scale • Participate with advancements in machine learning and apply them to continuously improve products Skills • MS/PhD in Computer Science, Electrical Engineering, or related field • Strong research experience with familiarity in top conferences (e.g., CVPR, ICCV, NeurIPS) • 5+ years of experience in Python and proficiency in Java, C++, or Scala • Strong understanding of diffusion models • Strong understanding of multi-threading and memory management • Solid knowledge of ML architectures: CNNs and Transformers • Experience with PyTorch or TensorFlow • Experience building end-to-end ML deployment and inference systems, especially for low-latency, real-time applications • Experience deploying ML models in cloud environments (AWS preferred) • Experience with experiment tracking systems and ML workflows • Experience in low level optimisation, cuda etc • Experience productionizing and scaling ML models in real-world systems • Contributions to open-source projects • Experience with MLOps tools or distributed training systems • Familiarity with relational databases (Postgres/MySQL) • Experience handling large-scale data using tools like Spark Benefits • 5 weeks of paid leave to recharge • Comprehensive healthcare benefits including vision and dental • Visa Support: We provide visa assistance, including H1B and OPT transfers, for US employees to ensure a smooth transition and support your career with us Company Overview • Reka AI is an AI research and product company that develops multimodal AI solutions to advance science and build generative AI models. It was founded in 2022, and is headquartered in Sunnyvale, California, USA, with a workforce of 11-50 employees. Its website is https://reka.ai. Company H1B Sponsorship • Reka AI has a track record of offering H1B sponsorships, with 2 in 2025, 1 in 2023. Please note that this does not guarantee sponsorship for this specific role. Apply To This Job Apply To this Job Company : frontendnode-production.up.railway.app Salary : Work from home