ML Engineer - NY
π Location: New York (hybrid)
π° Salary: $140,000 - $180,000 + stock options + benefits
About the Roleπ
We are a forward-thinking team of machine learning engineers developing specialized generative AI models. We collaborate with organizations worldwide, providing AI-powered solutions such as crisis text response assistance, all while curating top-tier datasets to fuel our models.βοΈ
We value empowering our engineers by offering them autonomy, flexibility, and resources to excel. If you're passionate about working in a fast-paced, early-stage start-up environment, we'd love to have you join us.π
Overviewπ
As the Lead ML Engineer, youβll partner closely with the founders and existing ML team to drive research, data acquisition, and model training from inception to completion. Youβll be responsible for training large-scale language models, working primarily with models with extensive parameter counts. The role includes fine-tuning models, optimizing based on user preferences, and keeping up with cutting-edge research in reinforcement learning for language models. You will be responsible for reading research papers, extracting state-of-the-art methodologies, and integrating them into our development process.π«°
Ideal Candidate π
- Experience in setting up and training language models on multi-GPU infrastructures.
- 6+ years working with deep learning frameworks like PyTorch, TensorFlow, or JAX, with at least 3+ years in a production environment.
- Proven track record in machine learning, with a strong ability to innovate or improve existing ideas, ideally demonstrated through first-author publications or notable projects.
- Strong knowledge of software engineering best practices, with an appreciation for high-quality, maintainable code.
- Ability to work in a fast-paced, results-oriented environment, focusing on quick MVPs over lengthy presentations.
- Effective at communicating technical concepts to non-technical stakeholders.
- Passionate about the transformative potential of deep learning.
Bonus Points:π―
- Experience working across Sales, Product, and Engineering teams in a high-growth STARTUP environment.
- Previous engineering experience in Data Engineering, LLM's, Gen AI, or cloud platforms like AWS/GCP/Azure.
Why Join?π²πβοΈ
- Innovate and Lead: Be part of a company thatβs pioneering the next generation of AI.
- Global Impact: Work with a team of industry veterans shaping the future of machine learning.
- Career Growth: Enjoy opportunities to advance your career in a rapidly growing company.π©