Futo Logo

Machine Learning Engineer

FUTO Keyboard is a source-first and privacy-first software keyboard for Android. Every prediction inference routine runs entirely on-device; no user data can ever leave the device. We support over 150 keyboard layouts and languages, and over 700 layout + language combinations (not all of them include dictionary or predictions)

Swipe typing is an important part of the mobile typing experience for many. FUTO Keyboard currently supports swipe typing across arbitrary layouts and languages by matching words over a dictionary with a greedy algorithm. Applying machine learning here poses a unique challenge due to the requirement of needing to support so many layouts and languages, as well as user-added personal dictionary words (or even layout customizations!), with limited data. We have previously collected and publicly released a US English QWERTY dataset: https://huggingface.co/datasets/futo-org/swipe.futo.org/

We are looking for a machine-learning researcher or engineer to work through these challenges and help us improve the swipe typing experience in FUTO Keyboard.

Your responsibilities will include:

  • Writing dataset processing pipelines, synthetic data generation, or other techniques as necessary
  • Designing, training and validating neural and/or probabilistic models for improved swipe typing, including language models
  • Designing within the limitations of low-end devices with limited storage, memory, processing power
  • Working together to help integrate efficient on-device inference in C++ (ggml)
  • Continue supporting swipe across arbitrary layout and language combinations via a fallback algorithm, a model that’s invariant to layout/dictionary, or something else

Required Qualifications:

  • 3+ years of professional experience in machine learning.
  • Demonstrated record of designing, training and shipping production ML models
  • Solid grasp of sequence-modelling techniques (RNNs, LSTMs, CTC, Transformers)
  • Experience with low-end optimization (quantization, pruning, distillation)
  • Comfort with collaborative software practices (Git)
  • FUTO is a process light organization that is largely driven by engineering. We expect you to be highly independent in this role.

Location:

FUTO is based in Austin, Texas. This position can be hybrid onsite or fully remote. We can offer relocation assistance to Austin.

Benefits:

  • Medical and dental insurance
  • 401k matching
  • Bonuses based on company performance

To apply, send an e-mail with your resume to: [email protected]