jax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
- Category
- Python packages
pip install jax-cuda13-plugin Official docs outline
jax-cuda13-plugin.readthedocs.ioExtra Credit builds an outline of jax-cuda13-plugin's docs the first time your work uses it. Until then, start at the official documentation.
More in Python packages
All 9,553-
mo-imports More Imports! - Delayed importing
-
threaded Decorators for running functions in Thread/ThreadPool/IOLoop
-
free-proxy Proxy scraper for further use
-
cocotb-bus
-
cmeel-octomap cmeel distribution for OctoMap, An Efficient Probabilistic 3D Mapping Framework Based on Octrees
-
audiomentations A Python library for audio data augmentation. Inspired by albumentations. Useful for machine learning.
-
pytorch-msssim Fast and differentiable MS-SSIM and SSIM for pytorch.
-
pysnc Python SNC (REST) API
-
durabletask A Durable Task Client SDK for Python
-
dask-image Distributed image processing
-
livekit-plugins-azure Agent Framework plugin for services from Azure
-
oic Python implementation of OAuth2 and OpenID Connect