Package Usage : pypi : jax
Explore the latest package usage data for jax in the pypi ecosystem, including detailed information on package dependencies and their usage.
Total dependents: 392
Total downloads: 1,084,879,634
More details on packages.ecosyste.ms - JSON
datasense/datasense ab9c0b165911027efbf61e52b46d8b50
- version: 0.4.14
tensorflow/tensorflow 2.18.0
Official Docker images for the machine learning framework TensorFlow (http://www.tensorflow.org)
- version: 0.4.5, 0.4.6
rasa/rasa 3.6.21
- version: 0.4.13
tensorflow/build 2.20-python3.9
Containers for building TensorFlow, provided by TensorFlow SIG Build.
- version: 0.3.25
anyscale/ray-ml 2.38.0
- version: 0.3.25, 0.4.4, 0.4.13
google/deepvariant 1.8.0
- version: 0.4.13
tensorflow/tfx 1.16.0
- version: 0.4.8
apache/beam_python3.8_sdk 2.60.0
Apache Beam SDK - Python 3.8
- version: 0.4.8
huggingface/transformers-pytorch-gpu 4.41.3
Hugging Face Transformers repository with CPU & GPU PyTorch backend
- version: 0.3.4
apache/beam_python3.9_sdk 2.59.0
- version: 0.4.8
huggingface/transformers-doc-builder latest
- version: 0.3.4
huggingface/transformers-pytorch-cpu 4.18.0
Hugging Face Transformers repository with CPU-only PyTorch backend
- version: 0.3.4
mirekphd/ml-cpu-py38-jup-cust 20230623
- version: 0.4.8
huggingface/transformers-tensorflow-gpu 4.41.3
- version: 0.3.4
mirekphd/ml-gpu-py38-cuda112-cust 20230623
- version: 0.4.8
huggingface/transformers-all-latest-gpu latest
- version: 0.3.4
huggingface/transformers-cpu 4.18.0
- version: 0.3.4
singlecellopenproblems/openproblems-python-pytorch latest
- version: 0.4.6
jcoffi/cluster-anywhere cu124
This is a work in progress and it doesn't work yet. It is based on Ray.io, mesh VPN, and CrateDB.
- version: 0.4.13
ultralytics/ultralytics 8.3.5
Ultralytics YOLOv8 Docker Images
- version: 0.4.18
huggingface/transformers-tensorflow-cpu 4.18.0
- version: 0.3.4
huggingface/transformers-gpu 4.18.0
- version: 0.3.4
huggingface/transformers-pytorch-deepspeed-latest-gpu latest
- version: 0.3.4
singlecellopenproblems/openproblems-python-bedtools latest
- version: 0.4.6
huggingface/transformers-all-latest-gpu-push-ci latest
- version: 0.3.4
bengal1/pyhf-funcx 0.6.0
- version: 0.2.13
oawofolu/image-processor-default b1.20240815.175531
- version: 0.4.8
secretflow/secretflow-anolis8 1.11.0b1
anolis 8 based image with secretflow installed
- version: 0.4.12, 0.4.1
xprobe/xorbits v0.8.2
Docker image repo for https://github.com/xprobe-inc/xorbits, powered by XProbe Inc.
- version: 0.3.25
huggingface/transformers-all-latest-torch-nightly-gpu latest
- version: 0.3.4
jinahub/xbhnid8l 643cdcaec469a1f3c22c7a65
- version: 0.3.7
huggingface/transformers-pytorch-deepspeed-latest-gpu-push-ci latest
- version: 0.3.4
trinkle23897/envpool-release 2023-01-02-5f1a5fd
envpool pypi wheel release docker image
- version: 0.3.6
huggingface/transformers-pytorch-deepspeed-nightly-gpu latest
- version: 0.3.4
oawofolu/ml-image-processor latest
- version: 0.4.12
yvonneng/plottingscript_withminisample v6
- version: 0.2.22
mshaikh/alphafold v2.1
- version: 0.2.14, 0.2.25
inseefrlab/onyxia-python-tensorflow py3.12.9-gpu-2025.05.26
- version: 0.4.8
luisas/af2_msa 2
- version: 0.3.17
nasanccs/helio-hackweek2020 latest
Docker container based on RAPIDS for NASA Helio Hackweek 2020
- version: 0.1.75
luisas/af2_split 2
- version: 0.3.17
adsoft/ai-ops latest
- version: 0.4.2
jiaodong/alpa v9
Hub to save images for Alpa project https://github.com/alpa-projects/alpa
- version: 0.3.22
bcanfieldsherman/agent eviq
- version: 0.3.25, 0.4.8
deeptimhe/ubuntu18.04-cuda11.3.1-python3.7-pytorch1.12 orig-sing-amlt9.9.2-pytorch3d0.7.0-tensorflow1.15
FROM: https://github.com/Lightning-AI/lightning/blob/master/dockers/base-cuda/Dockerfile
- version: 0.3.23
lemarkus2/asp-project cuda
;)
- version: 0.3.22
jihu1011/mobilenerf-tf latest
- version: 0.4.1
huggingface/transformers-pytorch-deepspeed-latest-gpu-test latest
- version: 0.3.4
huggingface/transformers-all-latest-gpu-test latest
- version: 0.3.4
mirekphd/ml-cpu-py39-vsc-cust 20231008
- version: 0.4.6
thefivetoes/bau-v1 test32
- version: 0.3.14
nfcore/proteinfold_colabfold 1.1.0
- version: 0.3.25
tsphan/cardial-nyu paperspace-xdgrasp-bionic-3.18.6-20221220c
- version: 0.4.1
nfcore/proteinfold_alphafold2_standard 1.1.0
- version: 0.3.17, 0.3.25
ericecht/vit-base-ic 01
- version: 0.3.4
tjmjaspers/rarp v9
- version: 0.4.6
rgabdullin/environment v1.1
- version: 0.2.24
thubregtsen/clockbox latest
- version: 0.4.14
polusai/basic-flatfield-correction-plugin 1.3.2
- version: 0.3.25
orcuna/libra4 latest
- version: 0.4.8
huggingface/transformers-pytorch-gpu-test latest
- version: 0.3.4
livetechprove/loko-vision latest
- version: 0.4.8
huggingface/transformers-pytorch-deepspeed-nightly-gpu-test latest
- version: 0.3.4
etretiakov/workbench-session-complete jammy-2024.09.19-custom-12.9
- version: 0.4.4, 0.4.1
kaizhang/scatac-bench 0.2.1
- version: 0.4.14
huggingface/transformers-all-latest-torch-nightly-gpu-test latest
- version: 0.3.4
lukaszkucinski/cuda118_cudnn86 latest
- version: 0.2.25
shl198/sc_ppl 202306
- version: 0.4.10
ufo101/kata-go-visualizer latest
- version: 0.4.10
ckiddo74/orbit_ml latest
- version: 0.1.77
acheng416/alphafold latest
- version: 0.2.14
cryolite/kanachan latest
- version: 0.4.4
harubaru1/model-serializer 4
- version: 0.3.15
huggingface/transformers-pytorch-past-1.9-gpu latest
- version: 0.3.4
anarhist/evaluation_task 0545708
- version: 0.3.14
aiarena/arenaclient-bot-base v0.7.0
- version: 0.4.10
gcriotfxosspublic/tfx 1.16.0
- version: 0.4.10
zerlinwang/motif changedefaultpython
- version: 0.2.12
aiarena/arenaclient-bot v0.6.10
- version: 0.4.10
mirekphd/ml-scraper-py39-jup 20231011
- version: 0.4.10
mirekphd/ml-cpu-py39-jup-lite 20230728
- version: 0.4.10