jax-js
JAX in JavaScript – ML library for the web, running on WebGPU & Wasm
File Explorer
Download Latest Version (.zip)- ci.yaml
- argreduce.bench.ts
- cholesky-webgpu.bench.ts
- fft.bench.ts
- matmul-wasm.bench.ts
- matmul-webgpu.bench.ts
- matvec-wasm.bench.ts
- matvec-webgpu.bench.ts
- sort.bench.ts
- svd.bench.ts
- index.ts
- opfs.ts
- safetensors.test.ts
- safetensors.ts
- tokenizers.test.ts
- tokenizers.ts
- weightMapper.ts
- package.json
- README.md
- convolution.ts
- elementwise.ts
- linalg.ts
- movement.ts
- reduction.ts
- utility.ts
- index.ts
- model.test.ts
- ops.ts
- tensor.ts
- package.json
- README.md
- alias.ts
- base.ts
- combine.ts
- index.ts
- losses.ts
- transform.ts
- treeUtils.ts
- adamw.test.ts
- clipByGlobalNorm.test.ts
- combine.test.ts
- sgd.test.ts
- transform.test.ts
- treeUtils.test.ts
- package.json
- README.md
- convert_to_safetensors.py
- optimize_onnx.py
- allocator.test.ts
- allocator.ts
- builtins.test.ts
- builtins.ts
- codegen.ts
- featureProbe.ts
- parallel.ts
- tilePlan.ts
- translation.ts
- wasmblr.test.ts
- wasmblr.ts
- builtins.ts
- builtins.ts
- codegen.ts
- firefoxPoll.ts
- nullaryKernel.ts
- reader.ts
- routines.ts
- tracing.ts
- backend.test.ts
- cpu.ts
- wasm.ts
- webgl.ts
- webgpu.ts
- array.test.ts
- array.ts
- convolution.test.ts
- convolution.ts
- core.ts
- jaxpr.ts
- jit.ts
- jvp.ts
- linearize.ts
- vmap.ts
- dtype-info.ts
- einsum.test.ts
- einsum.ts
- lax-linalg.ts
- lax.ts
- nn.ts
- numpy-fft.ts
- numpy-linalg.ts
- numpy.ts
- random.ts
- scipy-special.ts
- alu.test.ts
- alu.ts
- backend.ts
- index.ts
- polyfills.ts
- pprint.test.ts
- pprint.ts
- routine.ts
- shape.test.ts
- shape.ts
- tracing.ts
- tree.test.ts
- tree.ts
- tuner.ts
- utils.ts
- deno.test.ts
- node.test.ts
- argreduce.test.ts
- basic.test.ts
- conv.test.ts
- devices.test.ts
- dtype-f16.test.ts
- dtype-f64.test.ts
- dtype.test.ts
- lax-linalg.test.ts
- lax.test.ts
- nn.test.ts
- numpy-fft.test.ts
- numpy-linalg.test.ts
- numpy.test.ts
- random.test.ts
- refcount.test.ts
- scipy-special.test.ts
- setup.ts
- tracing.test.ts
- vitest.d.ts
- benchmarks.svg
- chat.svg
- detection.svg
- fallback.svg
- fluid.svg
- heat.svg
- mnist.svg
- mobileclip.svg
- nca.svg
- pca.svg
- tts.svg
- whisper.svg
- logo.svg
- format.ts
- LineChart.svelte
- DownloadManager.svelte
- DownloadToast.svelte
- Seo.svelte
- mnist.ts
- DemoThumbnail.svelte
- MatmulPerfDemo.svelte
- ConsoleLine.svelte
- EmbeddedRepl.svelte
- encode.ts
- ReplEditor.svelte
- runner.svelte.ts
- benchmark.ts
- monaco.ts
- profiling.ts
- +page.svelte
- strategies.ts
- +page.svelte
- strategies.ts
- +page.svelte
- strategies.ts
- +layout.svelte
- +page.svelte
- BenchmarkPage.svelte
- +page.svelte
- chat-model.ts
- gemma.ts
- huggingface-tokenizer.ts
- lfm.ts
- MarkdownMessage.svelte
- +page.svelte
- +page.svelte
- coco.ts
- +page.svelte
- +page.svelte
- +page.ts
- +page.svelte
- +page.svelte
- +page.svelte
- books.ts
- clipInference.ts
- +page.svelte
- +page.svelte
- +page.svelte
- 01-arrays-jit.ts
- 02-broadcasting.ts
- 03-reference-counting.ts
- 04-devices.ts
- 05-jit-static-args.ts
- 06-gradients.ts
- 07-make-jaxpr.ts
- 08-vmap.ts
- 09-jvp-vjp.ts
- 10-hessian.ts
- 11-tree-gradients.ts
- 12-pca-svd.ts
- 13-matmul-einsum.ts
- 14-least-squares.ts
- 15-cholesky-sampling.ts
- 16-solve-det-lu.ts
- 17-inverse-gradient.ts
- 18-sorting-topk.ts
- 19-fft-filter.ts
- 20-reaction-diffusion.ts
- 21-frequency-peaks.ts
- 22-low-pass-filter.ts
- 23-image-denoise.ts
- 24-convolution-theorem.ts
- 25-spectrogram.ts
- 26-conv1d.ts
- 27-edge-detection.ts
- 28-max-pooling.ts
- 29-batched-conv.ts
- 30-mandelbrot.ts
- 31-heat-equation.ts
- 32-tiny-neural-net.ts
- 33-logistic-regression.ts
- 34-mlp-classifier.ts
- 35-adam-vs-sgd.ts
- 36-gradient-clipping.ts
- 37-activations.ts
- 38-softmax.ts
- 39-attention.ts
- 40-causal-mask.ts
- 41-keys-split.ts
- 42-vmap-sampling.ts
- 43-monte-carlo-pi.ts
- 44-multivariate-normal.ts
- 45-categorical-gumbel.ts
- 46-tokenizer.ts
- 47-safetensors-keys.ts
- 48-weight-mapper.ts
- repl-builtins.d.ts
- +page.svelte
- audio.ts
- inference.ts
- pocket-tts.ts
- +page.svelte
- decoding.ts
- features.ts
- model.ts
- tokenizer.ts
- +error.svelte
- +layout.svelte
- +layout.ts
- +page.svelte
- app.css
- app.d.ts
- app.html
- hooks.server.ts
- social-image.png
- favicon.ico
- favicon.svg
- .gitignore
- eslint.config.js
- package.json
- svelte.config.js
- tsconfig.json
- vite.config.ts
- .editorconfig
- .gitignore
- .prettierignore
- eslint.config.ts
- FEATURES.md
- LICENSE
- package.json
- pnpm-lock.yaml
- pnpm-workspace.yaml
- README.md
- tsconfig.json
- tsdown.config.ts
- typedoc.ts
- vercel.json
- vitest.config.ts
# Installation Guide
1. Get the code
git clone https://github.com/ekzhang/jax-js
Downloads the entire project code from GitHub to your computer.
cd jax-js
Moves into the project folder you just downloaded.
2. Node.js
Easy RecommendedPrerequisites
npm i @jax-js/jax
Type this command into your terminal and run it.
pnpm install
Downloads and installs the libraries listed in package.json.
pnpm run build:watch
Builds optimized production files.
pnpm exec playwright install
Type this command into your terminal and run it.
pnpm test
Type this command into your terminal and run it.
After running the command, open the address shown in the terminal (usually something like http://localhost:3000) in your browser.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)