Stability AI · 4 hours ago

Senior Machine Learning Engineer, Integrity

United States

Full-time

Remote

Senior Level

5+ years exp

Maximize your interview chances

Artificial Intelligence (AI)Generative AI

Insider Connection @Stability AI

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

Responsibilities

Lead efforts to drive the design development and production of ML systems, and present the solutions to the leadership teams and customers

Work with the Research team on developing the next generation of safe models, where you may assist with areas such as optimization of model training, model tuning, dataset engineering, HPC clusters, tooling, and work on open-source efforts

Be a strategic thought partner for leaders across the organization on driving business impact through machine learning

Prototype and productionize model architecture improvements and new features enhancing our overall safety valves.

Writing and maintaining production-quality code, ensuring robustness, efficiency, and scalability to support the safety of our models in a live environment

Please note that this role may involve exposure to explicit content and topics, including those of a sexual, violent, or psychologically disturbing nature

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

Machine LearningPythonPyTorchJupyter NotebooksJAXTPUsCUDAJavaScriptLarge Language ModelsTrust & Safety

Required

Master or PhD in computer science or relevant areas with 5+ years experience working on machine learning projects, including training, fine tuning and refining models

Experience with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks

Ability to communicate machine learning concepts and results effectively through writing and visualization

Excellent communication skills with a proven ability to articulate technical challenges and solutions to varied audiences

Strong problem-solving ability and the ability to work in ambiguous and constantly evolving environment