Open role
[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
Company H1B Sponsorship