Deep learning

Current implementation of Go binding can not specify options.

GPUOptions struct is in internal package. And go generate doesn't work for protobuf directory. So we can't specify GPUOptions for NewSession.

Model description

I would like to add a new model:

Proposed in the paper: UNETR: Transformers for 3D Medical Image Segmentation

UNEt TRansformers (UNETR) utilize a transformer as the encoder to learn sequence representations of the input volume and effectively capture the global multi-scale information, while also following the successful "U-shaped"

Followup to pytorch/pytorch#74955 (comment).

It turns out that that cmake version was just bad and we can now unpin cmake once again.

cc @seemethere @malfet @pytorch/pytorch-dev-infra

What happened + What you expected to happen

The autoscaler pushes some logs to the Ray driver. The logs are prefixed with (scheduler) which is misleading.
The prefix should be (autoscaler).

Versions / Dependencies

Ray master.

Reproduction script

Submit a Ray task or actor which triggers upscaling to trigger the logs.

Issue Severity

Low: It annoys or frustrates me.