JotaiGPU Logo

JotaiGPU — a toolkit for building declarative data graphs, spanning across your CPU and GPU, with automatic dependency tracking. It’s built on top of Jotai and TypeGPU, allowing your entire data pipeline to be written in TypeScript, both the CPU and GPU portions.

Let’s look at an interactive example.

// Vanilla atoms can be decorated with a WGSL schema, allowing
// their value to be uploaded to the GPU
const
const countAtom: WritableAtom<number, [SetStateAction<number>], void> & WithUpload<d.F32>
countAtom
=
withUpload<d.F32, number>(schema: d.F32, wrappedAtom: WritableAtom<number, [SetStateAction<number>], void>): WritableAtom<number, [SetStateAction<number>], void> & WithUpload<...> (+1 overload)
withUpload
(
import d
d
.
const f32: d.F32
export f32

A schema that represents a 32-bit float value. (equivalent to f32 in WGSL)

Can also be called to cast a value to an f32.

@example const value = f32(true); // 1

f32
,
atom<number>(initialValue: number): PrimitiveAtom<number> & WithInitialValue<number> (+4 overloads)
atom
(1));
// GPU atoms compute their value on the GPU. They can derive
// other GPU atoms, or `withUpload(...)` decorated atoms
const
const doubleAtom: GPUAtom<d.F32, number>
doubleAtom
=
gpuAtom<d.F32>(schema: d.F32): GPUAtomConstructor<d.F32>
gpuAtom
(
import d
d
.
const f32: d.F32
export f32

A schema that represents a 32-bit float value. (equivalent to f32 in WGSL)

Can also be called to cast a value to an f32.

@example const value = f32(true); // 1

f32
)(() => {
'kernel'; // <- this function is executable on the GPU
return
const countAtom: WritableAtom<number, [SetStateAction<number>], void> & WithUpload<d.F32>
countAtom
.
WithUpload<F32>.$: number
$
* 2;
});
// GPU atoms are just asynchronous atoms from the perspective
// of vanilla atoms, so we can download the result and derive it
const
const quadAtom: Atom<Promise<number>>
quadAtom
=
atom<Promise<number>>(read: Read<Promise<number>, never>): Atom<Promise<number>> (+4 overloads)
atom
(async (
get: Getter
get
) => {
return (await
get: <Promise<number>>(atom: Atom<Promise<number>>) => Promise<number>
get
(
const doubleAtom: GPUAtom<d.F32, number>
doubleAtom
)) * 2;
});
// Listening to changes of `quadAtom`. There are also
// easy bindings for frameworks like React
const store: PrdOrDevStore
store
.
sub: (atom: AnyAtom, listener: () => void) => () => void
sub
(
const quadAtom: Atom<Promise<number>>
quadAtom
, async () => {
var console: Console

The console module provides a simple debugging console that is similar to the JavaScript console mechanism provided by web browsers.

The module exports two specific components:

  • A Console class with methods such as console.log(), console.error() and console.warn() that can be used to write to any Node.js stream.
  • A global console instance configured to write to process.stdout and process.stderr. The global console can be used without importing the node:console module.

Warning: The global console object's methods are neither consistently synchronous like the browser APIs they resemble, nor are they consistently asynchronous like all other Node.js streams. See the note on process I/O for more information.

Example using the global console:

console.log('hello world');
// Prints: hello world, to stdout
console.log('hello %s', 'world');
// Prints: hello world, to stdout
console.error(new Error('Whoops, something bad happened'));
// Prints error message and stack trace to stderr:
// Error: Whoops, something bad happened
// at [eval]:5:15
// at Script.runInThisContext (node:vm:132:18)
// at Object.runInThisContext (node:vm:309:38)
// at node:internal/process/execution:77:19
// at [eval]-wrapper:6:22
// at evalScript (node:internal/process/execution:76:60)
// at node:internal/main/eval_string:23:3
const name = 'Will Robinson';
console.warn(`Danger ${name}! Danger!`);
// Prints: Danger Will Robinson! Danger!, to stderr

Example using the Console class:

const out = getStreamSomehow();
const err = getStreamSomehow();
const myConsole = new console.Console(out, err);
myConsole.log('hello world');
// Prints: hello world, to out
myConsole.log('hello %s', 'world');
// Prints: hello world, to out
myConsole.error(new Error('Whoops, something bad happened'));
// Prints: [Error: Whoops, something bad happened], to err
const name = 'Will Robinson';
myConsole.warn(`Danger ${name}! Danger!`);
// Prints: Danger Will Robinson! Danger!, to err

@seesource

console
.
Console.log(message?: any, ...optionalParams: any[]): void

Prints to stdout with newline. Multiple arguments can be passed, with the first used as the primary message and all additional used as substitution values similar to printf(3) (the arguments are all passed to util.format()).

const count = 5;
console.log('count: %d', count);
// Prints: count: 5, to stdout
console.log('count:', count);
// Prints: count: 5, to stdout

See util.format() for more information.

@sincev0.1.100

log
(await
const store: PrdOrDevStore
store
.
get: <Promise<number>>(atom: Atom<Promise<number>>) => Promise<number>
get
(
const quadAtom: Atom<Promise<number>>
quadAtom
));
});