Perplexity Logo

Perplexity

AI Training Infrastructure Engineer - Post Training

Posted 10 Hours Ago
Be an Early Applicant
7 Locations
Senior level
7 Locations
Senior level
Build and implement a comprehensive post-training framework for large language models with a focus on scaling and integration into products.
The summary above was generated by AI

Perplexity is seeking experienced AI Research Engineers and Scientists to continue to improve our in house Online LLMs, the Sonar models. Your job is to work with team and create a robust and effective training framework (on top of Megatron/PyTorch), especially for post training LLMs.

Responsibilities

  • Build a post training framework that can run cutting-edge model training jobs in scale
  • Implement the necessary infra and components to support latest models and algorithms like SFT, RL (DPO/GRPO) and more
  • Own the full stack data, training, and eval pipelines required to post-train LLM models
  • Work closing with engineering teams to integrate Sonar models into our product.

Qualifications

  • Proven experience with large-scale LLMs frameworks building
  • Strong in Python/Pytorch; C++/CUDA is a plus
  • Self-starter with a willingness to take ownership of tasks
  • Passion for tackling challenging problems
  • Minimum of 6 years of working on relevant projects.

Bonus

  • PhD in AI/ML/Systems or related areas
  • Experience building LLM training frameworks, especially post training

The cash compensation range for this role is $220,000 - $290,000.

At Perplexity, we've experienced tremendous growth and adoption since publicly launching the world's first fully functional conversational answer engine in 2022. We've grown from answering 2.5 million questions per day at the start of 2024 to around 20 million daily queries in December 2024. We also offer Perplexity Enterprise Pro, which counts leading companies like Nvidia, the Cleveland Cavaliers, Bridgewater, and Zoom as customers.

To support our rapid expansion, we've raised significant funding from some of the most respected technology investors. Our investor base includes IVP, NEA, Jeff Bezos, NVIDIA, Databricks, Bessemer Venture Partners, Elad Gil, Nat Friedman, Daniel Gross, Naval Ravikant, Tobi Lutke, and many other visionary individuals. In 2024, our employee base grew nearly 300%, and we're just getting started.

Final offer amounts are determined by multiple factors, including, experience and expertise, and may vary from the amounts listed above.

Equity: In addition to the base salary, equity is part of the total compensation package.
Benefits: Comprehensive health, dental, and vision insurance for you and your dependents. Includes a 401(k) plan.


Top Skills

C++
Cuda
Python
PyTorch

Similar Jobs

2 Hours Ago
Remote
Hybrid
9 Locations
Senior level
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
CrowdStrike seeks a Senior Software Engineer to develop testing tools and frameworks for their cloud services, ensuring quality and reliability. Responsibilities include verifying functionality, improving automated test coverage, and maintaining test infrastructure.
Top Skills: Aws Ec2Aws S3C#C++CassandraDockerElasticsearchGoJavaJenkinsJSONKafkaKubernetesMemcacheMySQLNode.jsPostgresPythonRedisRestRpcScalaSqlserverXML
2 Hours Ago
Hybrid
Toronto, ON, CAN
Mid level
Mid level
Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
Seeking a Full Stack Angular Developer to create web apps, backend services, and mobile apps, using AWS and data mining. Collaboration with design teams and documentation required.
Top Skills: AngularAWSCapacitorCordovaGitNode.jsPlaywrightTypescript
3 Hours Ago
Hybrid
Toronto, ON, CAN
Senior level
Senior level
Fintech • Machine Learning • Payments • Software • Financial Services
Lead teams of mobile engineers to develop solutions that enhance customer experience and reduce fraud. Mentor engineers and drive technological innovation.
Top Skills: AndroidAWSGCPiOSAzure

What you need to know about the Ottawa Tech Scene

The capital city of Canada and the nation's fourth-largest urban area, Ottawa has proven a rapidly growing global tech hub. With over 1,800 tech companies, many of which are leaders in their sectors, the city's tech talent now makes up more than 13 percent of its total workforce. This growth is driven not only by the big players like UL Solutions and Dropbox, but also by a thriving startup ecosystem, as new businesses emerge to follow in the footsteps of those that came before them.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account