├── .github └── workflows │ ├── build.yml │ └── post-release.yml ├── .gitignore ├── .gitmodules ├── .npmignore ├── CMakeLists.txt ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── install.js ├── lib ├── core.ts ├── index.ts ├── nn │ ├── index.ts │ ├── init.ts │ ├── layers │ │ ├── activations.ts │ │ ├── base.ts │ │ ├── containers.ts │ │ ├── convolution-transpose.ts │ │ ├── convolution.ts │ │ ├── dropout.ts │ │ ├── embedding.ts │ │ ├── index.ts │ │ ├── linear.ts │ │ ├── normalization.ts │ │ ├── pooling.ts │ │ ├── positional-encoding.ts │ │ ├── pytools.ts │ │ ├── quantized.ts │ │ ├── recurrent.ts │ │ ├── transformer.ts │ │ └── upsample.ts │ ├── losses.ts │ └── utils.ts ├── optimizers │ ├── index.ts │ ├── optimizers.ts │ └── schedulers.ts └── utils.ts ├── node_mlx.node.d.ts ├── package.json ├── src ├── array.cc ├── array.h ├── bindings.cc ├── bindings.h ├── complex.cc ├── complex.h ├── constants.cc ├── device.cc ├── device.h ├── fast.cc ├── fft.cc ├── indexing.cc ├── indexing.h ├── io.cc ├── linalg.cc ├── memory.cc ├── metal.cc ├── ops.cc ├── ops.h ├── random.cc ├── stream.cc ├── stream.h ├── transforms.cc ├── trees.cc ├── trees.h ├── utils.cc └── utils.h ├── tests ├── array.spec.ts ├── autograd.spec.ts ├── compile.spec.ts ├── eval.spec.ts ├── fast.spec.ts ├── fft.spec.ts ├── init.spec.ts ├── js.spec.ts ├── linalg.spec.ts ├── losses.spec.ts ├── memory.spec.ts ├── nn.spec.ts ├── ops.spec.ts ├── optimizers.spec.ts ├── prompt_test_py_to_js.txt ├── random.spec.ts ├── run.ts ├── tree.spec.ts ├── tsconfig.json ├── utils.ts └── vmap.spec.ts └── tsconfig.json /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/post-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/.github/workflows/post-release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/.gitmodules -------------------------------------------------------------------------------- /.npmignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/.npmignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/README.md -------------------------------------------------------------------------------- /install.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/install.js -------------------------------------------------------------------------------- /lib/core.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/core.ts -------------------------------------------------------------------------------- /lib/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/index.ts -------------------------------------------------------------------------------- /lib/nn/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/index.ts -------------------------------------------------------------------------------- /lib/nn/init.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/init.ts -------------------------------------------------------------------------------- /lib/nn/layers/activations.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/activations.ts -------------------------------------------------------------------------------- /lib/nn/layers/base.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/base.ts -------------------------------------------------------------------------------- /lib/nn/layers/containers.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/containers.ts -------------------------------------------------------------------------------- /lib/nn/layers/convolution-transpose.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/convolution-transpose.ts -------------------------------------------------------------------------------- /lib/nn/layers/convolution.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/convolution.ts -------------------------------------------------------------------------------- /lib/nn/layers/dropout.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/dropout.ts -------------------------------------------------------------------------------- /lib/nn/layers/embedding.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/embedding.ts -------------------------------------------------------------------------------- /lib/nn/layers/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/index.ts -------------------------------------------------------------------------------- /lib/nn/layers/linear.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/linear.ts -------------------------------------------------------------------------------- /lib/nn/layers/normalization.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/normalization.ts -------------------------------------------------------------------------------- /lib/nn/layers/pooling.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/pooling.ts -------------------------------------------------------------------------------- /lib/nn/layers/positional-encoding.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/positional-encoding.ts -------------------------------------------------------------------------------- /lib/nn/layers/pytools.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/pytools.ts -------------------------------------------------------------------------------- /lib/nn/layers/quantized.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/quantized.ts -------------------------------------------------------------------------------- /lib/nn/layers/recurrent.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/recurrent.ts -------------------------------------------------------------------------------- /lib/nn/layers/transformer.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/transformer.ts -------------------------------------------------------------------------------- /lib/nn/layers/upsample.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/layers/upsample.ts -------------------------------------------------------------------------------- /lib/nn/losses.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/losses.ts -------------------------------------------------------------------------------- /lib/nn/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/nn/utils.ts -------------------------------------------------------------------------------- /lib/optimizers/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/optimizers/index.ts -------------------------------------------------------------------------------- /lib/optimizers/optimizers.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/optimizers/optimizers.ts -------------------------------------------------------------------------------- /lib/optimizers/schedulers.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/optimizers/schedulers.ts -------------------------------------------------------------------------------- /lib/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/lib/utils.ts -------------------------------------------------------------------------------- /node_mlx.node.d.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/node_mlx.node.d.ts -------------------------------------------------------------------------------- /package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/package.json -------------------------------------------------------------------------------- /src/array.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/array.cc -------------------------------------------------------------------------------- /src/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/array.h -------------------------------------------------------------------------------- /src/bindings.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/bindings.cc -------------------------------------------------------------------------------- /src/bindings.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/bindings.h -------------------------------------------------------------------------------- /src/complex.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/complex.cc -------------------------------------------------------------------------------- /src/complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/complex.h -------------------------------------------------------------------------------- /src/constants.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/constants.cc -------------------------------------------------------------------------------- /src/device.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/device.cc -------------------------------------------------------------------------------- /src/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/device.h -------------------------------------------------------------------------------- /src/fast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/fast.cc -------------------------------------------------------------------------------- /src/fft.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/fft.cc -------------------------------------------------------------------------------- /src/indexing.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/indexing.cc -------------------------------------------------------------------------------- /src/indexing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/indexing.h -------------------------------------------------------------------------------- /src/io.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/io.cc -------------------------------------------------------------------------------- /src/linalg.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/linalg.cc -------------------------------------------------------------------------------- /src/memory.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/memory.cc -------------------------------------------------------------------------------- /src/metal.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/metal.cc -------------------------------------------------------------------------------- /src/ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/ops.cc -------------------------------------------------------------------------------- /src/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/ops.h -------------------------------------------------------------------------------- /src/random.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/random.cc -------------------------------------------------------------------------------- /src/stream.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/stream.cc -------------------------------------------------------------------------------- /src/stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/stream.h -------------------------------------------------------------------------------- /src/transforms.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/transforms.cc -------------------------------------------------------------------------------- /src/trees.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/trees.cc -------------------------------------------------------------------------------- /src/trees.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/trees.h -------------------------------------------------------------------------------- /src/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/utils.cc -------------------------------------------------------------------------------- /src/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/src/utils.h -------------------------------------------------------------------------------- /tests/array.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/array.spec.ts -------------------------------------------------------------------------------- /tests/autograd.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/autograd.spec.ts -------------------------------------------------------------------------------- /tests/compile.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/compile.spec.ts -------------------------------------------------------------------------------- /tests/eval.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/eval.spec.ts -------------------------------------------------------------------------------- /tests/fast.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/fast.spec.ts -------------------------------------------------------------------------------- /tests/fft.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/fft.spec.ts -------------------------------------------------------------------------------- /tests/init.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/init.spec.ts -------------------------------------------------------------------------------- /tests/js.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/js.spec.ts -------------------------------------------------------------------------------- /tests/linalg.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/linalg.spec.ts -------------------------------------------------------------------------------- /tests/losses.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/losses.spec.ts -------------------------------------------------------------------------------- /tests/memory.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/memory.spec.ts -------------------------------------------------------------------------------- /tests/nn.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/nn.spec.ts -------------------------------------------------------------------------------- /tests/ops.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/ops.spec.ts -------------------------------------------------------------------------------- /tests/optimizers.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/optimizers.spec.ts -------------------------------------------------------------------------------- /tests/prompt_test_py_to_js.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/prompt_test_py_to_js.txt -------------------------------------------------------------------------------- /tests/random.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/random.spec.ts -------------------------------------------------------------------------------- /tests/run.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/run.ts -------------------------------------------------------------------------------- /tests/tree.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/tree.spec.ts -------------------------------------------------------------------------------- /tests/tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/tsconfig.json -------------------------------------------------------------------------------- /tests/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/utils.ts -------------------------------------------------------------------------------- /tests/vmap.spec.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tests/vmap.spec.ts -------------------------------------------------------------------------------- /tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/frost-beta/node-mlx/HEAD/tsconfig.json --------------------------------------------------------------------------------