Grantham Taylor

SNOW-960172: Add `map_dict` functionality

1

## What is the current behavior? I have some arbitrary values in a column. Let's say the unique values of this column are in the set `{"foo", "bar", "buzz"}`. I...

[BUG] `TensorDict` and `TensorClass` objects are not supported in `torch.jit.fork`

1

## Describe the bug These objects cannot be outputted from forked functions. This is extremely unfortunate because, well, tensordicts are such a great way to handle multiple, heterogeneous blocks of...

Grantham/add-attach_shm-template

5

## Why are the changes needed? Adding SHM is a necessity for multi-GPU ML training workloads. It is currently not immediately obvious how to do this. ## What changes were...