Lead the design, development, and optimization of ML systems using generative models. Collaborate with research to improve models and productionize solutions.
Senior Machine Learning Engineer, Applied - United States or Canada
About the role:
We are looking for Machine Learning engineers passionate about generative models and the creative applications of AI to join our Applied ML team. We look for highly innovative ML engineers with a strong blend of research and engineering skills who are motivated to push the boundaries of image, video, and 3D foundation models and their applications. In this role, you will collaborate with research and platform teams to build, improve, and optimize workflows that power state-of-the-art application solutions based on diffusion models. You will have access to high-performance computing resources and work alongside top researchers and engineers, making a real impact in the rapidly growing field of generative AI.
Responsibilities:
- Lead efforts to drive the design, develop, prototype and productionize ML systems that address the customer needs
- Collaborate with the research team on developing the next generation of models, where you may assist with areas such as optimization of model training, model fine-tuning, dataset engineering, tooling, open-source efforts, etc.
- Work on the commercial side, productionizing generative model-based solutions, by partnering with platform and research teams building the infrastructure to serve them at scale
- Write production-quality code and influence the next generation of Stability AI’s machine learning solutions
- Advocate for engineering best practices and upskill the team
Qualifications:
- MS or PhD in Computer Science or related technical discipline
- +3 years working on machine learning projects
- Experience working with diffusion models
- Proficiency with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks
- Strong coding skill and comfortable with coding in production platforms
- Ability to diagnose technical problems, debug code, and automate manual routine tasks
- Ability to communicate machine learning concepts and results effectively through writing and visualization
- Track record of publication in top-tier machine learning conferences
Top Skills
Colab
Jupyter
Python
PyTorch
Similar Jobs
Marketing Tech • Real Estate • Software • PropTech • SEO
As a Growth Staff Engineer, you will drive lead generation strategies, optimize funnels, mentor engineers, and champion innovative growth solutions.
Top Skills:
ApolloAWSDynamoDBElasticsearchKubernetesLambdaNode.jsPostgresPythonReactSinglespaStyled ComponentsTypescript
Consumer Web • Enterprise Web • Mobile • Productivity • Software
As a Senior Backend Engineer, you'll design and develop backend systems that support revenue growth, collaborate across teams, and ensure code quality while working with various technologies.
Top Skills:
GoGCPKubernetesPostgresRedisStripe
Big Data • Fintech • Mobile • Payments • Financial Services
The role involves developing machine learning models for credit opportunity management, collaborating with cross-functional teams, and optimizing data pipelines at Affirm.
Top Skills:
KubeflowNumpyPandasPythonRayScikit-LearnSparkSQLXgboost
What you need to know about the Austin Tech Scene
Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.
Key Facts About Austin Tech
- Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
- Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
- Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
- Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
- Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center
