🧰 JAX

The items that name it 4

• Hugging Face ❤️ TorchTPU: TPUs, Natively
  articleas frameworks and tools2026-10-20

  Framework around which the TPU software stack grew, alongside TensorFlow.

• TPU Inference Externalization Full Steam Ahead - InferenceX
  articleas frameworks and tools2026-09-07

  Framework used to execute models on TPU, with jax.jit capturing inference steps.

• GRADSOLVE: fast exact gradients for ODE ensembles on GPUs
  paperas classification.technologies2026-09-02
• rohitg00/ai-engineering-from-scratch
  courseas frameworks and tools

  Framework introduced in Phase 3 lesson 12.

In every workspace you can read 5