All Katas - ML Katas

Implementing One-Hot Encoding with `scatter_`

this month easy (<10 mins) | scatter one-hot encoding

Your task is to create a function that performs one-hot encoding on a tensor of integer labels. This is a common preprocessing step for categorical data in machine learning. You will be given a 1D...

Implementing a simplified Beam Search Decoder with `gather` and `scatter`

this month hard (>1 hr) | NLP scatter gather

Beam search is a popular decoding algorithm used in machine translation and text generation. A key step in beam search is to select the top-k most likely next tokens and update the corresponding...

Creating a Label Mapping with `scatter` for Domain Adaptation

this month easy (<30 mins) | scatter indexing

In domain adaptation, you might need to map labels from a source domain to a target domain. Imagine you have a set of labels and a mapping that specifies how each old label corresponds to a new...

Efficiently Updating a Sub-Tensor with `scatter_`

this month hard (<1 hr) | scatter indexing

Sometimes you need to update a portion of a larger tensor with values from a smaller tensor, where the update locations are specified by an index tensor. This is a common pattern in various...