Machine Learning Engineer

Wordbricks, Inc.; San Francisco (CA); Remote; USD 140,000 - 210,000; Full time

Benefits offered by this job

Remote option

SF Bay Area presence

Seoul office

Tokyo office

Annual Company Retreat

Comprehensive Insurance Coverage

Retirement Benefits

Job summary

Wordbricks, Inc. is seeking a Machine Learning Engineer to design and implement NLP models powering our productivity tools.

You will optimize inference speed and deploy ML pipelines across training and deployment stages, collaborating with frontend and backend teams to integrate AI features. We value hands-on experience with deep learning, NLP, and large language models, plus strong Python skills and production-ready ML systems knowledge.

Qualifications

• 3+ years of experience in machine learning engineering.
• Strong background in deep learning and natural language processing.
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow).
• Experience with large language models and their applications.

Responsibilities

• Design and implement ML models for various natural language processing tasks.
• Optimize model performance and inference speed for production environments.
• Develop and maintain ML pipelines for training and deployment.
• Collaborate with frontend and backend teams to integrate AI features.

Skills

NLP

Deep Learning

Python

PyTorch

TensorFlow

LLMs

Model deployment

Tools

PyTorch

TensorFlow

Python

Job description

SF Bay Area or Remote Offices in Seoul and Tokyo

What we're looking for

As a Machine Learning Engineer at Wordbricks, you will be at the forefront of developing and implementing AI solutions that make complex tasks simple and intuitive for users. You'll work on creating and optimizing ML models that power our next-generation productivity tools.

Responsibilities

• Design and implement ML models for various natural language processing tasks
• Optimize model performance and inference speed for production environments
• Develop and maintain ML pipelines for training and deployment
• Collaborate with frontend and backend teams to integrate AI features

Qualifications

• 3+ years of experience in machine learning engineering
• Strong background in deep learning and natural language processing
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow)
• Experience with large language models and their applications

Preferred Qualifications

• Experience with LLM fine-tuning and prompt engineering
• Knowledge of ML systems design and scalable inference
• Background in software engineering and production systems

About Wordbricks

At Wordbricks, we believe AI shouldn't be locked away for tech experts—it should be for anyone who's ever sighed at cluttered files, copy-pasted the same thing 47 times, or spent hours wrangling messy data.

We're a group of curious minds pushing limits and making work smarter. Our organization is flat, and our team is small and talent-dense. We particularly value creativity, passion, and a drive to make complex technology accessible to everyone.

Why Join Wordbricks?

• Be part of a dynamic and forward-thinking team dedicated to technological innovation.
• Access to cutting-edge technology and resources to support your professional growth.
• Competitive compensation package and benefits.
• A collaborative and inclusive work culture that values diversity and creativity.
• SF Bay Area or Remote - your choice; offices in Seoul and Tokyo
• Flexible Work Schedule
• Annual Company Retreat
• Comprehensive Insurance Coverage and Retirement Benefits