chunkax. A lightweight JAX transform for applying functions to array chunks over arbitrary sizes and dimensions.

github.com/alebeck/chunkax
by @alebeck

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.