Cox Exponential Logo

Cox Exponential

Research Scientist/Engineer, Efficient ML Systems

Reposted 21 Days Ago
Remote or Hybrid
Hiring Remotely in CA, USA
Mid level
Remote or Hybrid
Hiring Remotely in CA, USA
Mid level
Research and build efficient ML systems for large-scale LLMs and agentic RL: design algorithms and system techniques, prototype in training/inference stacks, run large-scale experiments, and translate findings into production or publications.
The summary above was generated by AI
About Goaly

At Goaly, our mission is to make custom AI affordable for every business. Our founding team comes from the front lines of top AI labs and tech giants (Meta MSL, TikTok AI, Google DeepMind, xAI, Microsoft Research, etc.), where we built large-scale training infrastructure powering trillion-parameter models and scaled GenAI models to a global user base. Now, we are building something we wish we had before: a platform that makes training and adapting custom AI affordable for all modern companies, not just Big Tech. Our north star is ambitious: for a domain-specific task, reach 90% of SOTA performance at less than 10% of the cost. To get a taste of what we are doing, see our first tech blog.


About the Role

As an AI Research Scientist (Efficient ML Systems) at Goaly, you will research and build the systems that make frontier-scale models practical. This role sits at the intersection of algorithms, systems, and hardware efficiency.You will design and evaluate new training and inference techniques, prototype them in real systems, and push them to production-scale workloads.

Your work will either ship directly into our core platform or lead to publications at top venues such as NeurIPS, ICML, ICLR, or CVPR.This is not a paper-only role. You will write real systems code, run large-scale experiments, and directly shape how modern LLMs and RL systems are trained and deployed.


Core Responsibilities

  • Research efficient ML systems: Invent and evaluate algorithms and system techniques that improve LLM and agentic RL training and inference efficiency (memory, compute, communication, and stability).
  • Scale agentic RL: Design and optimize large-scale agentic RL pipelines, including asynchronous training, experience management, reward modeling, and long-horizon stability.
  • End-to-end experimentation: Design large-scale experiments spanning model architecture, training algorithms, distributed systems, and hardware-aware optimization.
  • System-aware research: Prototype research ideas directly in training and inference stacks (e.g., parallelism strategies, attention kernels, RL training pipelines) and validate them at scale.
  • Production & publication: Translate successful ideas into production-ready systems and/or publish them at top-tier conferences with full internal support.

Requirements

  • Ph.D. or Master's degree in CS, AI, Systems, or related fields (Exceptional undergraduates with strong research capabilities may be considered).
  • Strong foundation in LLM or large-scale ML training, including Transformers, attention mechanisms, distributed training, and optimization methods.
  • Experience or strong interest in agentic RL or large-scale reinforcement learning systems, including stability, scalability, or long-horizon training challenges.
  • Demonstrated interest in efficiency-focused research, such as training acceleration, memory optimization, parallelism, kernels, or RL system robustness.
  • Proficient in PyTorch or JAX. Clean coding style and strong command of Python.
  • Adaptability: A fast learner with a strong sense of responsibility, capable of wearing multiple hats and handling cross-stack challenges.


Bonus Points

  • First-author publications at top conferences (NeurIPS, ICML, ICLR, CVPR, ACL).
  • High-star open-source projects on Hugging Face or Gold/Silver medals in Kaggle competitions.


Similar Jobs

20 Minutes Ago
Easy Apply
Remote
Easy Apply
140K-170K Annually
Senior level
140K-170K Annually
Senior level
Cloud • Security • Software • Cybersecurity • Automation
Lead product direction for tenant scale and unified deployment across GitLab.com, Dedicated, and Self-Managed. Define requirements for tenant portability, scalability, isolation, and regional compliance. Partner with engineering, product peers, and customers to translate infrastructure constraints into product strategy, prioritize tradeoffs, and align distributed teams to deliver a unified deployment platform.
Top Skills: AutoscalingCloud ServicesGitlabHorizontal ScalabilityMultitenancySaaS
An Hour Ago
Easy Apply
Remote
Easy Apply
154K-154K Annually
Junior
154K-154K Annually
Junior
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
Design, build, and maintain backend services for fraud and risk detection at scale. Deliver features end-to-end, partner with Data Science/ML, improve reliability and observability, respond to incidents, and build reusable systems to reduce financial loss and protect users.
Top Skills: AWSBitcoinDockerDynamoDBEthereumGCPGeminiGleanGoJavaKafkaKubernetesLibrechatMongoDBPostgresPythonRabbitMQRuby
6 Hours Ago
Remote or Hybrid
US
128K-145K Annually
Expert/Leader
128K-145K Annually
Expert/Leader
Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Manage regional field inspection and audit operations to meet budgets and customer requirements. Oversee staffing, performance management, continuous improvement, client technical engagement, and adherence to UL policies and procedures. Drive customer satisfaction, subcontractor relationships, and implementation of preventive/corrective actions.

What you need to know about the NYC Tech Scene

As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.

Key Facts About NYC Tech

  • Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
  • Key Industries: Artificial intelligence, Fintech
  • Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
  • Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account