Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
7 daysStars36.6k24h gain+16Score6.8
Search
Find trending repositories by name or description.
8 results for “tpu”
RankRepository7 daysStars24h gainScore
Moving to vllm-project/vllm-jax. TPU inference for vLLM, with unified JAX and PyTorch support.
7 daysStars47924h gain0Score0.2
7 daysStars16924h gain0Score0.0
Backend.AI is a streamlined, container-based computing cluster platform that hosts popular computing/ML frameworks and diverse programming languages, with pluggable heterogeneous accelerator support including CUDA GPU, ROCm GPU, Gaudi NPU, Google TPU, GraphCore IPU and other NPUs.
7 daysStars67624h gain-Score0.0
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
7 daysStars2.8k24h gain-Score0.0
Machine learning compiler based on MLIR for Sophgo TPU.
7 daysStars99824h gain-Score0.0
Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
7 daysStars19424h gain-Score0.0
Optimize Julia Functions With MLIR and XLA for High-Performance Execution on CPU, GPU, TPU and more.
7 daysStars38924h gain0Score0.0