Applied AI Engineer & Researcher

Description

About the company

At the company (the company's site/), we're building a new model for deterministic developer tasks that expects high accuracy like OCR, web scraping, classification, STT, and more. We currently power thousands of systems in industries like healthcare, finance, government, SaaS and more which has processed billions of tokens every month.

We're hiring an AI Engineer & Researcher who is looking to push the boundaries of our models by implementing and experimenting with new research and techniques. As part of the the company lab, you will help improve the performance and capabilities of our models by managing our AI lifecycle, including fine-tuning, automating data collection and cleaning, benchmarking, and deployment.

As a founding member at the company, your role will be dynamic, offering opportunities to spearhead and launch innovative research, papers, and products into production while being the go-to expert for all things AI!

Responsibilities

• Research and understand different machine learning techniques and papers, then experiment with real-world implementation
• Experience with deep learning frameworks (e.g., TensorRT-LLM, PyTorch), and AI development tools is a must
• Working closely with the team to train, deploy and serve models at scale on popular cloud providers like AWS, GCP, and more
• Write papers on your research and experiments contributing to the growing open-source AI world
• Writing detailed benchmarks on comparisons and analysis (e.g. comparing performance for two embedding models or benchmarking whisper 3 with another model)
• Quantization and fine-tuning of LLMs and other models for our developer use cases
• Optimizing models and GPU infra for cost-to-scale ratio based on load
• Writing Python code and Jupyter Notebooks

Requirements

• A strong AI Researcher with at least 5 years of experience, and paper publications at venues like ICML, NeurIPS, ACL, NAACL, and IEEE
• Worked with most of our GPU infra and have a good understanding of our tech stack
• Comfortable with working across the stack - model deployment, benchmarking, fine-tuning, monitoring, and scaling the service
• Up-to-date with the latest papers, techniques, and experiments
• Using AI to 100x your productivity
• Natural curiosity to experiment with new concepts and tools while working closely with our community of developers
• You care about building great products and love to geek out about the latest tech in AI
• Open to feedback and have a growth mindset
• Bonus if you
• Enjoy working in a startup environment where you can take initiative and contribute to the zero-to-one phase of development
• Previously ran a startup or worked as a founding team member in a startup
• Experience: 1+ years
• Visa: Will sponsor

Benefits

• Great compensation package and equity/shares
• Flexible and remote-friendly
• Annually run off-sites
• Learn and Grow - we provide mentorship and send you to events that help you build your network and skills
• Annual Education Allowance
• Let's build benefits together
• Process
• The entire process is fully remote and all communication will happen over email or via video chat.
• Once you've submitted your application, the team will review your submission and may reach out for a short screening interview over a video call.
• If you pass the screen you may be invited to complete a short task or discuss your past work.
• If you pass the task/screen you will be invited to up to 2 follow-up interviews. The calls:
• usually take between 20-45 minutes each depending on the interviewer.
• are all 1:1.
• will be with both founders, a member of either the growth or engineering team (depending on the role) and usually one other person from your immediate team or function.