Match score not available

Data Scientist (LLM)

Work set-up:

Full Remote

Contract:

Full time

Work from:

Brazil

Offer summary

Qualifications:

Strong understanding of Large Language Models, including transformer internals and tokenization., Solid foundation in statistics, machine learning, and optimization., Proficiency in Python, with experience in PyTorch or JAX., Experience training models beyond notebooks, including multi-GPU and distributed setups..

Key responsibilities:

Design, train, and optimize LLMs for product features and internal tools.
Analyze model performance, debug issues, and implement improvements.
Collaborate with engineering and product teams to enhance model capabilities.
Work on architecture, memory efficiency, and scaling of LLMs.

CloudWalk, Inc. SME https://www.cloudwalk.io

201 - 500 Employees

See all jobs

Job description

Who we are:

CloudWalk is a fintech company reimagining the future of financial services. We are building intelligent infrastructure powered by AI, blockchain, and thoughtful design. Our products serve millions of entrepreneurs across Brazil and the US every day, helping them grow with tools that are fast, fair, and built for how business actually works. Learn more at cloudwalk.io .

Who We’re Looking For:

We’re hiring a Data Scientist focused on Large Language Models to join our AI R&D team. You’ll work on designing, training, and optimizing LLMs that power real product features and internal tools. You’ll touch everything from architecture and data preparation to multiscale training and evaluation.

This is not a prompt engineering or RAGfocused role. We’re looking for someone who gets the math, understands the models, and knows how to write code that scales.

You’ll work sidebyside with engineers, MLOps, and product teams to make our LLMs smarter, faster, and more useful.

What Youll Do: Train and finetune LLMs using multiGPU and distributed setups.
Analyze model performance, debug failures, and implement improvements.
Work with curated and synthetic datasets, including classification and generation tasks.
Design experiments, track results, and iterate quickly with tools like MLflow.
Write clean, productionready code and collaborate via GitHub.
Push boundaries: think about architecture, memory efficiency, scale, and cost.

What We’re Looking For:
LLM understanding: You know how these models work under the hood transformer internals, tokenization, embeddings, etc.
Stats and ML fundamentals: You have a solid foundation in statistics, machine learning, and optimization.
Coding skills: You write Python well. You’ve worked with PyTorch or JAX. You don’t fear Bash or git.
Training experience: You’ve trained models beyond notebooks. Bonus if you’ve worked with mixed precision, DeepSpeed, or multinode training.
Systems mindset: You understand tradeoffs throughput vs memory, latency vs accuracy.
Good judgment: You know when to read a paper, when to read a stack trace, and when to rewrite the dataloader or part of the LLM architecture.

Bonus Points:
Experience with Hugging Face Transformers, Datasets, Accelerate.
Familiarity with Kubernetes, Ray, or custom training infra.
Exposure to embeddings, classification tasks, or tokenlevel losses.
Ability to mentor or guide junior researchers or engineers.

How We Hire:
Online assessment: Online assessment: technical logic and fundamentals (MathCalculus, Statistics, Probability, Machine LearningDeep Learning, Code).
Technical interview: dive into theory and reasoning (no code).
Cultural interview.

If you are not willing to take an online quiz, do not apply.
Diversity and inclusion:We believe in social inclusion, respect, and appreciation of all people. We promote a welcoming work environment, where each CloudWalker can be authentic, regardless of gender, ethnicity, race, religion, sexuality, mobility, disability, or education.