#tensorflow #bindings

wasm-tensorflow-tfjs-core

Rust WASM Bindings for the npm package @tensorflow/tfjs-core

2 releases

0.1.1 Dec 7, 2023
0.1.0 Nov 30, 2023

#726 in WebAssembly

MIT license

4KB

wasm-tensorflow-tfjs-core

What if you wanted to do this in JavaScript?

import '@tensorflow/tfjs-backend-webgl'
import { setBackend } from '@tensorflow/tfjs-core'

await setBackend('wegbl')

Just add this crate and you can do it in Rust!

// Somehow import '@tensorflow/tfjs-backend-webgl'
use wasm_tensorflow_tfjs_core::{set_backend, BackendName};

let result = set_backend(BackendName::Webgl).await;

Dependencies

~1–2.7MB
~52K SLoC