The company is looking to advance its effort to create the best-performing open AI model by developing state-of-the-art models that excel in scientific discovery, reasoning, and factual recall.
Requirements
- Expertise at building ML infrastructure
- Deep experience in the complete model development cycle, including data set construction, training, tuning, evaluation, performance profiling, and monitoring
- Knowledge of modern deep learning and natural language processing techniques
- Strong software engineering skills, particularly around building performant systems and debugging
- Experience with Python and PyTorch/Jax/Tensorflow
- Familiarity working with cloud compute resources (e.g. AWS) and containerization (e.g. Docker)
- Contributions to open-source ML or research libraries (e.g. spaCy, AllenNLP, transformers)
Responsibilities
- Building infrastructure to facilitate the next generation of LLM research
- Optimizing training and inference for language models
- Triaging between experiments and executing on the most impactful
- Supporting and collaborating with an open-source community
- Bridging the gap between cutting-edge research and a widely adopted product
- Bringing software engineering best practices to a research environment
- Releasing contributions back to the broader community in the form of open source software, model releases, and additions to Ai2’s public API and open research datasets, as well as technical reports
Other
- BS or MSc in Computer Science, Statistics, Engineering, Applied Mathematics, or a related quantitative field
- Must be able to remain in a stationary position for long periods of time
- The ability to communicate information and ideas so others will understand
- The ability to observe details at close range
- Can work under deadlines
- Team members and their families are covered by medical, dental, vision, basic life insurance, basic accidental death and dismemberment insurance, short-term disability, long-term disability, and an employee assistance program
