jax
05614edc - Expose pallas.mosaic.random.sample_block to pltpu interface
Commit
View On GitHub
Commit
1 year ago
Expose pallas.mosaic.random.sample_block to pltpu interface PiperOrigin-RevId: 730923727
Parents
8ac29697
FAQ Terms Privacy Refunds Impressum
Loading