mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
4ba0c24a8f
* export and import functions * refactor + works for few primitives * nit * allow primitives with state * nit * nit * simplify serialize / deserialize * fix for constants * python bindings * maybe fix serialize failure case * add example * more primitives, training kind of works * same result for python and c++ * some fixes * fix export * template it up * some simplificatoin * rebase * allow kwargs and multiple functions * exporter * more primitives for exporting * deal with endianness * handle invalid stream * add docstring
22 lines
485 B
C
22 lines
485 B
C
// Copyright © 2023 Apple Inc.
|
|
|
|
#pragma once
|
|
|
|
#include "mlx/array.h"
|
|
#include "mlx/backend/metal/metal.h"
|
|
#include "mlx/compile.h"
|
|
#include "mlx/device.h"
|
|
#include "mlx/distributed/distributed.h"
|
|
#include "mlx/distributed/ops.h"
|
|
#include "mlx/einsum.h"
|
|
#include "mlx/export.h"
|
|
#include "mlx/fast.h"
|
|
#include "mlx/fft.h"
|
|
#include "mlx/io.h"
|
|
#include "mlx/linalg.h"
|
|
#include "mlx/ops.h"
|
|
#include "mlx/random.h"
|
|
#include "mlx/stream.h"
|
|
#include "mlx/transforms.h"
|
|
#include "mlx/utils.h"
|