elixir-nx/axon

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Flax is a neural network library for JAX that is designed for flexibility.

An Extensible Deep Learning Library

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

JAX-based neural network library

You like pytorch? You like micrograd? You love tinygrad! ❤️

Supercharge Your Model Training