Smarter Technology For All is seeking a Model Development Engineer to contribute to cutting-edge research and development in generative AI, specifically in training large language models, large vision models, and large multimodal models.
Requirements
- Strong programming skills in Python and experience with deep learning frameworks like PyTorch.
- Solid understanding of machine learning principles, including supervised learning, unsupervised learning, and reinforcement learning.
- Familiarity with large language models, transformer architectures, and related concepts.
- Experience with data processing tools and techniques (e.g., Pandas, NumPy).
- Experience working with Linux systems and/or HPC cluster job scheduling (e.g., Slurm, PBS).
- Experience with distributed training frameworks (e.g., DeepSpeed, Megatron-LM)
- Experience with PyTorch
Responsibilities
- Design, implement, and evaluate training pipelines for large generative AI models, encompassing multiple stages of post-training.
- Develop and implement adversarial training techniques to improve model robustness against adversarial attacks and enhance generalization performance by exposing the model to perturbed input examples during training.
- Develop and perform model distillation techniques to compress large language models into smaller, more efficient models while preserving key performance characteristics.
- Implement and evaluate model quantization techniques to reduce model size and accelerate inference speed, balancing precision loss with performance gains for deployment across diverse hardware platforms.
- Utilizing techniques for efficient fine-tuning of large language models, balancing performance and resource constraints, and tailoring model performance for downstream tasks well.
- Experiment with various training techniques, hyperparameters, and model architectures to optimize performance and efficiency.
- Develop and maintain data pipelines for processing and preparing training data.
Other
- Bachelor's or Master's degree in Computer Science, Machine Learning, or a related field and 5+ years of relevant work experience or 7+ years of relevant work experience.
- Ph.D. in Computer Science, Machine Learning, or a related field.
- Excellent communication, collaboration, and problem-solving skills.
- We are an Equal Opportunity Employer and do not discriminate against any employee or applicant for employment because of race, color, sex, age, religion, sexual orientation, gender identity, national origin, status as a veteran, and basis of disability or any federal, state, or local protected class.
- Collaborate with other engineers and researchers to design, implement, and deploy AI-powered products.
