[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.ResponsibilitiesTranslate cutting-edge research into production-ready machine learning systemsDesign, build, and deploy end-to-end ML models and pipelinesDevelop and optimize models for image and video processingOwn the full ML lifecycle: experimentation, training/fine-tuning, evaluation, and deploymentRapidly prototype using open-source models and adapt them for product needsConduct experiments, analyze results, and iterate to improve performanceCollaborate with researchers and cross-functional teams (product, engineering, design) to deliver ML solutions at scaleParticipate with advancements in machine learning and apply them to continuously improve productsSkillsMS/PhD in Computer Science, Electrical Engineering, or related fieldStrong research experience with familiarity in top conferences (e.g., CVPR, ICCV, NeurIPS)5+ years of experience in Python and proficiency in Java, C++, or ScalaStrong understanding of diffusion modelsStrong understanding of multi-threading and memory managementSolid knowledge of ML architectures: CNNs and TransformersExperience with PyTorch or TensorFlowExperience building end-to-end ML deployment and inference systems, especially for low-latency, real-time applicationsExperience deploying ML models in cloud environments (AWS preferred)Experience with experiment tracking systems and ML workflowsExperience in low level optimisation, cuda etcExperience productionizing and scaling ML models in real-world systemsContributions to open-source projectsExperience with MLOps tools or distributed training systemsFamiliarity with relational databases (Postgres/MySQL)Experience handling large-scale data using tools like SparkBenefits5 weeks of paid leave to rechargeComprehensive healthcare benefits including vision and dentalVisa Support: We provide visa assistance, including H1B and OPT transfers, for US employees to ensure a smooth transition and support your career with usCompany OverviewReka 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 SponsorshipReka 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.