rlx-mlx-sys 0.2.13

Low-level MLX C++ build + C ABI shim for RLX (vendored mlx, NVRTC-free).
1
2
3
4
5
6
7
8
9

# Copyright © 2023 Apple Inc.

from mlx.nn import init, losses
from mlx.nn.layers import *
from mlx.nn.utils import (
    average_gradients,
    fsdp_apply_gradients,
    value_and_grad,
)