Metal breaks jax.debug_callback

Graphics & Games General
GeoffNN OP
Created Feb ’24
Replies 0
Boosts 0
Views 382
Participants 1

When developping jax code locally, I use jax's debug_callback. Metal does not implement it.

NotImplementedError: MLIR translation rule for primitive 'debug_callback' not found for platform METAL

Replies  0
Boosts  0
Views  382
Participants  1
Metal breaks jax.debug_callback
First post date Last post date
 
 
Q