TL;DR
Member of Technical Staff (AI Engineering): Building and implementing state-of-the-art machine learning models at scale for generative media products with an accent on model performance and stability. Focus on deploying large neural network models to production environments, collaborating with researchers, backend engineers, and designers.
Location: Onsite in New York
Company
Ideogram’s mission is to make world-class design accessible to everyone, multiplying human creativity by building proprietary generative media models and AI native creative workflows.
What you will do
- Build and implement state of the art machine learning models at scale.
- Iteratively identify and implement model performance.
- Deploy machine learning models to production environments ensuring ongoing performance and stability.
- Collaborate with researchers, backend engineers and designers to deliver generative media products.
Requirements
- Experience with JAX, PyTorch or an equivalent deep learning framework.
- Experience in AI research, including training, fine-tuning, and experimenting with foundation models beyond black-box use.
- Experience in large neural network model production inference, including model parallelism.
- Experience with large accelerator cluster inference, including fleetwise optimizations.
- Requires a Master’s degree or foreign equivalent in Computer Science or a related field.
- At least one year of experience as a Machine Learning Engineer or related occupation.
Culture & Benefits
- Competitive compensation and equity.
- 4 weeks of vacation.
- Comprehensive health, vision, and dental coverage starting on day one.
- RRSP/401(k) with employer match up to 4%.
- Top-of-the-line tools and tech.
- A culture of learning and growth, where curiosity is encouraged and mentorship is part of the journey.