axon
Axon是一个基于Nx的Elixir深度学习库,提供高效的神经网络功能。核心组件包括数值定义的Functional API、模型创建的Model Creation API和训练模型的Training API。Axon通过模块化设计,实现各API独立使用,并通过Polaris进行优化。该库支持多种深度学习层和模型格式,如TensorFlow Lite和ONNX,适用于多种深度学习应用。