AI engineer
Build and deploy scalable AI applications and intelligent agents for enterprise customers. Requires 5+ years of experience with Python, ML frameworks (PyTorch/TensorFlow/JAX), LLMs, and cloud platforms.
About the job
What you'll do
- Architect, develop, and deploy high-performance, scalable AI applications into production environments, ensuring robust integrations with our end-to-end platform.
- Drive the development of intelligent agents and AI-powered features, translating complex research into practical, impactful solutions for our customers.
- Collaborate closely with research scientists, data scientists, and product managers to define technical requirements and deliver features that address critical business needs.
- Contribute to the research and evaluation of emerging AI technologies, frameworks, and tools to maintain WRITER's leadership in the enterprise generative AI space.
- Champion responsible AI practices, including bias detection, explainability, and safety alignment, making sure our AI is trustworthy and ethical.
What you need
- 5+ years of professional experience building and deploying machine learning or AI systems in a production environment.
- Proficiency in Python and experience with relevant AI/ML frameworks such as PyTorch, TensorFlow, or JAX.
- A strong foundation in machine learning principles, deep learning architectures, and natural language processing, particularly with large language models.
- Experience with cloud platforms (e.g., AWS, GCP, Azure) and MLOps tools for managing the AI lifecycle.
- Demonstrated ability to write clean, efficient, and well-tested code for scalable software systems.
- Exceptional problem-solving skills and a passion for tackling complex technical challenges with creative AI solutions.
Benefits & perks (US Full-time employees)
- Generous PTO, plus company holidays
- Medical, dental, and vision coverage for you and your family
- Paid parental leave for all parents (16 weeks)
- Fertility and family planning support
- Early-detection cancer testing through Galleri
- Flexible spending account and dependent FSA options
- Health savings account for eligible plans with company contribution
- Annual work-life stipends for wellness, learning and development
- Company-wide off-sites and team off-sites
- Competitive compensation, company stock options and 401k
Skills
Python, PyTorch, TensorFlow, JAX, Machine Learning, Deep Learning, NLP, LLMs, AWS, GCP, Azure, MLOps
Similar jobs
ML Engineering jobs
Applied Intuition
Sunnyvale, CA
Develop and deploy ML-first behavior prediction and planning systems for autonomous vehicles, forecasting the motion and interactions of road users. Requires a bachelor's degree, deep learning lifecycle expertise, and at least three years of production software experience with C++ or Python.
OnePay
United States
Build and operate production AI agents, automation workflows, and integrations that improve complex business processes. The role requires 5+ years of software engineering experience, modern LLM and agent-framework expertise, systems integration skills, and strong cross-functional collaboration.
LangChain
New York, NYBuild and teach reliable AI agent systems through customer workshops, technical content, guidance, and reference implementations. The role requires strong Python and agent-development experience plus a background delivering customer-facing technical training.
Roboflow
San Francisco, CA
Build reinforcement-learning environments, evaluations, datasets, and scalable infrastructure for frontier AI capabilities. The role suits a high-agency generalist engineer with experience in agents, evaluations, or RL workflows and strong communication skills.
Beacon Biosignals
Boston, MADevelop and productionize machine- and deep-learning algorithms for biosignal and EEG data used in medical devices, clinical development, and diagnostics. The role requires 4+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated environments and production ML practices.