Job Description
About Us
Nebula AI Systems is at the forefront of the generative AI revolution. We are building the infrastructure that powers the next generation of intelligent applications. We are seeking a visionary Senior AI/ML Engineer to join our elite engineering team and drive innovation in large language models (LLMs) and neural architecture search.
The Role
In this pivotal role, you will design, train, and deploy state-of-the-art machine learning models. You will work closely with our research scientists and product teams to translate cutting-edge academic research into production-ready software that impacts millions of users globally.
What You'll Do
Responsibilities
- Model Development: Design and implement novel deep learning architectures for natural language processing and computer vision.
- Training Pipelines: Build scalable, distributed training pipelines using frameworks like PyTorch and TensorFlow.
- Model Optimization: Apply techniques such as quantization, pruning, and distillation to optimize model performance and latency.
- Mentorship: Mentor junior engineers and data scientists, conducting code reviews and technical architecture discussions.
- Production Deployment: Collaborate with MLOps engineers to deploy models into robust, scalable cloud environments (AWS/GCP).
- Research: Stay abreast of the latest academic papers and industry trends to integrate new methodologies into our product suite.
Qualifications
- Education: Masterβs or PhD in Computer Science, Mathematics, or a related field with a focus on AI/ML.
- Experience: 5+ years of professional experience in machine learning engineering.
- Programming: Strong proficiency in Python, C++, and CUDA.
- Frameworks: Deep expertise in PyTorch or TensorFlow.
- Infrastructure: Experience with containerization (Docker) and orchestration (Kubernetes).
- Problem Solving: Proven track record of solving complex, unstructured problems in large-scale systems.