Research Engineer, Pre-Training
Jump Trading - New York, NY
Hiring: Research Engineer, Pre-Training Company: Jump Trading Location: New York, NY Job Posted Time: 2026-09-10 10:22:15 Target Skills & Keywords : ASIC, Data Pipeline, Deep Learning, FPGA, Fine-tuning, Kubernetes, Linear, Machine Learning, PyTorch, Python, Reinforcement Learning, Triton About the job Required Skills: •Expertise and track record of significant, measurable performance improvements in large-scale distributed training (MFU, throughput, convergence, cost-per-token). •Published research in efficient training methods, scaling laws, architectures, or systems for ML •Background in numerical computing, HPC, or distributed systems, including familiarity with GPUs/TPUs, high-performance networking (NVLink/InfiniBand), Kubernetes/Slurm, and OS internals •Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX) •Advanced degree (MS or PhD) in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier lab •Demonstrated capacity to balance ambitious research goals with practical engineering constraints •Strong problem-solving skills, results orientation, and excellent collaborative communication Qualifications: •Expertise in: CUDA kernel development, Triton/Pallas/CuTe DSLs, PyTorch/JAX internals, XLA optimization, or hardware acceleration (FPGA/ASIC) •Knowledge of reinforcement learning, post-training, or fine-tuning techniques •Knowledge of financial markets or trading •Annual Base Salary Range : $300,000 USD - $350,000 USD Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!