400Ping opened a new issue, #1156:
URL: https://github.com/apache/mahout/issues/1156
## Summary
Add a TPU backend for QDP using JAX (and Pallas for custom kernels), so
encoding workflows can run on Google TPU instead of being tied to CUDA-only
infrastructure.
## Primary Motivation
We want to avoid NVIDIA vendor lock-in.
Today, core execution paths are CUDA-centric, which limits deployment
flexibility, cost control, and hardware strategy.
## Scope
- Add a backend abstraction layer for device-specific implementations.
- Implement a `jax_tpu` backend for:
- amplitude encoding
- angle encoding
- basis encoding
- Add backend selection/configuration in API (`cuda`, `jax_tpu`, ...).
- Add correctness parity tests against current CUDA outputs (with
tolerances).
- Add docs for setup and backend selection.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
