Reka

Member of Technical Staff (Machine Learning Engineer)

Reka
Apply
5 months ago
Remote, WorldwideSenior
H1B sponsor

Responsibilities

  • Translate research into production-ready machine learning systems.
  • Design, build, deploy, evaluate, and optimize end-to-end ML models and pipelines.
  • Develop image and video processing models and adapt open-source models for product needs.
  • Own experimentation, training and fine-tuning, evaluation, deployment, and inference workflows.
  • Conduct experiments, analyze results, and improve model performance.
  • Collaborate with research, product, engineering, and design teams to deliver ML solutions at scale.

Requirements

  • Required: MS or PhD in Computer Science, Electrical Engineering, or a related field.
  • Required: strong research experience and familiarity with conferences such as CVPR, ICCV, and NeurIPS.
  • Required: 5+ years of Python experience and proficiency in Java, C++, or Scala.
  • Required: understanding of diffusion models, multithreading, memory management, CNNs, and Transformers.
  • Required: experience with PyTorch or TensorFlow and with end-to-end ML deployment and inference systems for low-latency, real-time applications.
  • Required: experience deploying ML models in cloud environments, preferably AWS, and using experiment tracking systems and ML workflows.
  • Preferred: low-level optimization and CUDA, productionizing and scaling ML models, open-source contributions, MLOps or distributed training, relational databases such as Postgres or MySQL, and large-scale data tools such as Spark.

Benefits

  • Remote-first work environment with a globally distributed team.
  • Five weeks of paid leave.
  • Comprehensive healthcare benefits, including vision and dental.
  • Visa assistance, including H1B and OPT transfers, for US employees.
  • Collaborative, inclusive, and mission-driven work environment.

Tech Stack

Categories

Reka

About Reka

51-200 employees

Reka is a privately held AI research and product company building multimodal foundation models and tools for organizations and businesses. Headquartered in Sunnyvale with a remote-first team, it offers products such as Reka Vision and Reka Edge, a 7B vision-language model. Founded by researchers from Google DeepMind and Meta’s FAIR, it also operates Claru, a data unit supplying licensed training datasets for physical and embodied AI.

Contact me