JetBrains

JetBrains

JetBrains provides cutting-edge development tools like IntelliJ IDEA and Kotlin, automating tasks to boost productivity and foster innovation.

Internet Software & Services
1K-5K
Founded 2000

Description

  • Work with stakeholders to translate business requirements into technical specifications.
  • Train large language models from scratch on a large GPU cluster.
  • Collect, process, and prepare pre-training and fine-tuning datasets.
  • Deploy models into production environments for global end-user access.
  • Support and improve existing subsystems.
  • Plan projects and make decisions independently when needed.
  • Take ownership of an entire subsystem.
  • Collaborate with others to identify customer needs and prioritize work accordingly.

Requirements

  • Experience in the design, deployment, and support of production ML systems.
  • Strong theoretical background in NLP and transformer-based approaches.
  • Proficiency with modern deep learning frameworks such as PyTorch and common NLP libraries.
  • Experience with distributed training of multi-billion-parameter models.
  • Attention to detail and strong communication skills.
  • Experience with LLM inference frameworks such as vLLM, DeepSpeed, or TensorRT (preferred).
  • Experience with LLM alignment techniques such as RLHF or RLAIF (preferred).
  • Experience with MLOps tools and practices, including CI/CD for ML (preferred).
  • Experience with Kubernetes and Kubeflow (preferred).
  • Scientific publications in the NLP field (preferred).
  • Passion for learning and staying current with the latest developments in the LLM field.

Benefits

  • Access to a large training cluster of hundreds of NVIDIA GPUs.
  • Use of modern ML tooling, including Python, PyTorch, HuggingFace, Kubeflow, Weights & Biases, Git, and TeamCity.
  • Opportunity to work on AI capabilities across JetBrains products.
  • An inclusive workplace that welcomes candidates regardless of background, identity, religion, age, accessibility needs, or orientation.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Staff Applied Research Engineer, Biometrics

Veriff 51-250 IT Services

Veriff’s Biometric Solutions team is hiring a computer vision applied researcher to accelerate the development of biometric and liveness defenses that protect the automated identity verification funnel.

Bash Computer Vision Machine Learning OpenCV Python PyTorch SQL TensorFlow
1 day, 12 hours ago

Staff AI Engineer - Notebooks

Datadog 5K-10K IT Services

Datadog is hiring a Staff Software Engineer to lead AI/ML initiatives for its Notebooks product, building intelligent workflows that support data analysis, investigations, and technical documentation.

Machine Learning
1 day, 12 hours ago

Senior Staff Engineer, AI

AlphaSense 251-1K Internet Software & Services

AlphaSense is hiring a Senior Staff AI Engineer to lead the design and implementation of AI systems that extract insights from millions of unstructured documents and multimedia files across its content intelligence platform.

CI/CD Computer Vision Django Docker FastAPI Kubernetes LLM Machine Learning MLOps Spring Boot
1 day, 14 hours ago

Research Intern - Reinforcement Learning, Self-Driving

Applied Intuition 251-1K Internet Software & Services

Applied Intuition is hiring multiple Research Interns to help advance next-generation physical AI through research in autonomous driving and robotics.

Computer Vision Machine Learning Reinforcement Learning
1 day, 16 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers