Machine Learning — PhD Intern (Dynamic I/O Schemas for Neural Models) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Keysight Technologies · 2 days ago

Machine Learning — PhD Intern (Dynamic I/O Schemas for Neural Models)

Keysight Technologies is at the forefront of technology innovation, delivering breakthroughs and trusted insights in electronic design and simulation. The PhD Intern in Machine Learning Systems will design and prototype runtime-adaptive neural models capable of modifying their input/output schemas dynamically, collaborating with experts to integrate these mechanisms into Keysight’s AI modeling stack.

AnalyticsCloud SecurityElectronicsManufacturingNetwork SecurityProduct DesignSoftwareTest and MeasurementWireless
badNo H1Bnote

Responsibilities

Design and implement dynamic I/O schema functionality in libtorch-based architectures, supporting runtime addition or removal of inputs and outputs
Develop modular components that maintain consistent state and gradient flow across evolving input/output configurations
Create schema translation and mapping utilities to maintain backward compatibility and incremental fine-tuning
Integrate schema management into training, checkpointing, and inference workflows
Benchmark adaptability and retraining efficiency, quantifying improvements in compute utilization and convergence time
Collaborate with runtime engineers to ensure performance, memory stability, and model safety under dynamic schema changes
Document and publish experimental results and architecture designs for internal and research dissemination

Qualification

C/C++LibtorchNeural architecturesDynamic computation graphsModular ML systemsSchema evolutionGPU optimizationAnalytical skillsSoftware engineeringAttention to efficiencyMemory managementVersion control

Required

Current PhD student (or recently graduated PhD) in Machine Learning, Computer Science, Applied Mathematics, or Electrical/Mechanical Engineering
Strong proficiency in C/C++ and libtorch (C++ PyTorch API) for neural network implementation
Understanding of dynamic computation graphs, model serialization, and runtime configuration management
Experience designing or training modular neural architectures or runtime-adaptive ML systems
Familiarity with schema evolution, metadata management, or flexible I/O processing
Strong analytical and software engineering skills with attention to efficiency, safety, and reusability
Experience designing and training GNN and GCN neural architectures

Preferred

Experience with dynamic-shape models using TorchScript, TensorRT, or ONNX Runtime
Background in graph- or operator-based architectures that support variable topologies
Understanding of parameter-efficient fine-tuning (PEFT), adapter layers, or meta-learning strategies
Experience profiling or optimizing GPU-based C++ inference and training pipelines

Company

Keysight Technologies

company-logo
Keysight Technologies is an electronic measurement company.

Funding

Current Stage
Public Company
Total Funding
$1.35B
Key Investors
Department for Science, Innovation and Technology (DSIT)
2025-04-10Post Ipo Debt· $750M
2024-10-02Post Ipo Debt· $600M
2023-09-15Grant

Leadership Team

leader-logo
Satish Dhanasekaran
President and Chief Executive Officer
linkedin
B
Brad Doerr
Vice President and General Manager
linkedin
Company data provided by crunchbase