JupyterLab + JAX

CUDA 12

SciPy notebook with JAX 0.11.2 and its CUDA 12 wheels.

PublishedjupyterNotebooks

Needs 7.5 GiB of GPU memory · CUDA on NVIDIA

Pick a machine

No machine can run this right now.

New machines appear as hosts connect them.

Have a GPU? Earn with it See all machines

Doesn't fit (4)

1× Apple M5 Max (40-core GPU) · New hostno inference runtime for this machine's backend1× Apple M4 Pro (20-core GPU) · New hostno inference runtime for this machine's backend1× Apple M4 · New hostno inference runtime for this machine's backend1× Apple M1 Pro (14-core GPU) · New hostno inference runtime for this machine's backend

Template details

Rating: No ratings yet
Runs: 0
Min VRAM: 8.1 GB
Backend: Set by the machine you pick
Version: jupyter-jax-v1.0.0
Image digest: 5546d5c51cf9…
Health check: SSH probe on the leased endpoint

We match GPUs on their measured runtime and dedicated VRAM. A hardware report doesn't qualify another vendor's backend.

#notebook #jax

Reviews

No reviews yet.