axon

收录理由

在 Elixir 的深度学习生态里，Axon 算得上是最贴近语言本身的选择。它直接构建在 Nx 之上，模型本质上就是数值定义，可以交给任意 Nx 后端或编译器执行，无论是 CPU、GPU 还是其他环境，甚至能导出成 ONNX 或 TensorFlow Lite 格式。整体设计分成三层：底层是涵盖激活函数、层和损失函数的函数式 API，中间是模型创建的高层接口，还有一套受 PyTorch Ignite 启发的训练 API，每一层都能独立使用。如果你已经在用 Nx 做张量运算，或者希望训练和推理都留在 Elixir 服务内部，不必另起 Python 进程，Axon 能让你只用一种语言。同时，这种解耦的结构也允许你插入自定义的数值定义或自写训练循环，同时继续利用内置的构建模块。