Principal AI and Machine Learning Engineer

• As a key contributor to AI/ML infrastructure initiatives, you will plan, execute, and analyze comprehensive benchmarks on switches, focusing on throughput, latency, congestion, incast, failover, path diversity, and workload performance to ensure optimal AI/ML network operations
• You will be guiding AI/ML workload deployments from initial scoping and test planning through execution and benchmark analysis, ensuring success criteria are met. Your role includes developing AI-driven automation workflows to streamline network development, operations, and implementations
• You will validate switch ASIC features including buffers, schedulers, QoS/queuing, ECMP behavior, telemetry, hashing, traffic distribution, and congestion visibility
• Owning switch OS configuration and automation, you will utilize SONiC, Junos, Ansible, Python, Bash, Git, and related tooling to implement and validate advanced features such as SRv6, segment routing, uSID, Adj-SID, and policy-based pathing as required. You will document PoC architecture, benchmark methodologies, topology diagrams, configurations, results, findings, and recommendations
• This role empowers you to shape the future of AI infrastructure networking by delivering scalable, high-performance, and resilient network fabrics that meet the stringent demands of AI/ML workloads, driving innovation and customer success at < >

Benefits

• The chance to earn an annual bonus based on performance
• A hybrid, flexible work model that ensures connectivity, a commitment to our culture and continued impact for team members
• Recognition for your contributions, and potential participation in long-term incentives like annual equity awards
• Subsidized health insurance, retirement programs and paid time off, including an industry-leading maternity and paternity leave package
• Opportunities for personal and professional development to help you reach your career goals- Docker or Kubernetes experience
• Python for automation experience
• Experience with Traffic tools such as Spirent, IXIA or similar
• Experience with L2/L3 network protocols such as BGP, OSPF, EVPN, VxLAN, IPv6 or similar
• Bachelors + 7 years of related experience, or Masters + 4 years of related experience
• Experience with network testing and validation
• Experience with RDMA, RoCEv2, PFC, ECN, congestion control, QoS, buffer behavior, and lossless Ethernet concepts
• Experience handling complex network segmentation, security policies, and multi-site fabric designs
• Deep understanding of Leaf-spine fabric and troubleshooting them
• Experience with Apstra and related automation tools for provisioning, managing and troubleshooting the fabric
• SONiC, Junos, Linux or other open source network operating systems experience
• Clear written and verbal communication skills as well as documentation skills