Skip to content

Add fused Metal cross entropy kernel - #4545

Closed
DIvkov575 wants to merge 2 commits into
ml-explore:mainfrom
DIvkov575:metal-cross-entropy-losses
Closed

Add fused Metal cross entropy kernel#4545
DIvkov575 wants to merge 2 commits into
ml-explore:mainfrom
DIvkov575:metal-cross-entropy-losses

Route nn.losses.cross_entropy to the fused kernel on Metal

996269b
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar