Machine Learning Engineer

CHAOS Inc. is a global technology startup that is delivering next-generation capabilities to the defense and critical industrial sectors. Founded in 2022 by a seasoned leadership team, CHAOS has quickly become the place where world-class multi-disciplinary engineers come to build mission-critical technologies. CHAOS has a mission-focused culture, dedicated to solving the toughest technical challenges. Its unique agile engineering approach enables rapid prototyping, while deep partnerships with defense and industry ensure real-world solutions.

Role Overview

Chaos AI division is seeking a highly skilled and motivated engineer to join our team. You will be responsible for developing, implementing, and maintaining our AI solutions related to vision, language, and other modalities. You will work closely with other engineers, data scientists, and product managers to design and build scalable and reliable AI-powered solutions that can process vast amounts of documents efficiently.

Responsibilities

1. Design and develop novel models that effectively combine information from vision and language models.
2. Research and explore the latest advancements in multimodal machine learning, LLMs and related fields through continuous learning and professional development activities.
3. Preprocess and curate datasets to ensure they are suitable for training the proposed models.
4. Evaluate and analyze model performance on various tasks and datasets to identify areas of improvement.
5. Optimize hyperparameters and model configurations to achieve the best performance across different modalities.
6. Apply transfer learning techniques to enhance model generalization and reduce training time.
7. Collaborate with cross-functional teams and develop APIs to integrate our technology into existing systems.
8. Assist in the deployment of trained models to production environments and monitor their performance to ensure reliability and scalability.
9. Work with product managers to define requirements and develop solutions that meet business needs.
10. Maintain detailed documentation of model designs, implementation, experiments, and results for knowledge sharing and future reference.
11. Work full-time onsite out of our office in Bellevue, Washington.

Minimum Requirements

1. Master's and/or PhD degree in Computer Science, Machine Learning, or a related field with at least 2 years professional experience in Machine Learning.
2. Solid understanding of machine learning fundamentals and experience with deep learning frameworks (e.g., PyTorch).
3. Extensive experience in training deep neural network models (such as training on multiple nodes/GPUs).
4. Strong programming skills in Python, C++, or similar languages and Linux scripting.
5. Ability to thrive and collaborate in ambiguous environments, working closely with team members to solve challenging technical problems.
6. Excellent communication skills and the ability to work in a team environment.
7. U.S. Person status (U.S. Citizen or permanent resident) is required as this position will require the ability to access US-only data systems and various sensitive projects.

Preferred Requirements

1. Experience with Computer Vision and Document Processing.
2. Experience with cloud platforms such as AWS, Azure, or Google Cloud for model development & deployment.
3. Familiarity/experience with Huggingface/LangChain/PyTorch Lightning.
4. Experience working with vision models.

Why CHAOS?

1. Our Benefits: Your medical, dental and vision benefits will be 100% paid for by the company.
2. Additional Benefits: life, FSA, HSA, 401k (+ Company match), and more.
3. Our Perks: free lunch every day, ‘No meeting Fridays’, unlimited PTO (for exempt employees), casual dress code, beautiful office location in Hermosa Beach with ocean views!
4. Compensation Components: competitive base salaries, generous pre-IPO stock option grants, relocation assistance + (coming soon!) annual bonuses.
5. Company Size: 45 employees and counting!
6. More About Us: We're building the next generation of aerospace, defense and critical industry technologies with a seasoned team of mission-focused individuals. We were founded in June 2022 and our goal is to become the platform where world-class engineers come to solve critical, real-world problems.

Salary range: $160k - $230k

The stated compensation range reflects only the targeted base compensation range and excludes additional earnings such as bonus, equity, and benefits. If your compensation requirements fall outside of the range, we still encourage you to apply. The salary range for this role is an estimate based on a range of compensation factors, inclusive of base salary only. Actual salary offer may vary based on (but not limited to) work experience, education and/or training, critical skills, and/or business considerations.