elixir-nx/nx

Nx-powered Neural Networks

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

A fast, ergonomic and portable tensor library in Nim with a deep learning focus for CPU, GPU and embedded devices via OpenMP, Cuda and OpenCL backends

TensorLy: Tensor Learning in Python.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.