FluxML/Flux.jl

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

JAX-based neural network library

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

An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

Deep Learning for humans

A WebGL accelerated JavaScript library for training and deploying ML models.

Minimalist ML framework for Rust