Software Engineer - Model Performance
Baseten- Full Time
- Mid-level (3 to 4 years), Senior (5 to 8 years)
Candidates must possess strong programming skills in Python and have experience with machine learning frameworks such as PyTorch, TensorFlow, or JAX. Hands-on experience with model optimization, quantization, and inference acceleration is required, along with a deep understanding of Transformer architectures and distributed inference techniques. Knowledge of quantization methods and memory-efficient inference techniques is essential, as well as a solid grasp of software engineering best practices including CI/CD and containerization technologies like Docker and Kubernetes.
The Machine Learning Engineer will design, build, and optimize machine learning deployment pipelines for large-scale models. They will implement and enhance model inference frameworks and develop automated workflows for model development, experimentation, and deployment. Collaboration with research, architecture, and engineering teams to improve model performance and efficiency is expected. The role also involves working with distributed computing frameworks to optimize model parallelism and deployment, implementing scalable KV caching and memory-efficient inference techniques, and monitoring and optimizing infrastructure performance across various levels of custom hardware.
AI compute platform for datacenters
d-Matrix focuses on improving the efficiency of AI computing for large datacenter customers. Its main product is the digital in-memory compute (DIMC) engine, which combines computing capabilities directly within programmable memory. This design helps reduce power consumption and enhances data processing speed while ensuring accuracy. d-Matrix differentiates itself from competitors by offering a modular and scalable approach, utilizing low-power chiplets that can be tailored for different applications. The company's goal is to provide high-performance, energy-efficient AI inference solutions to large-scale datacenter operators.