equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
AI summary
Neural network library
A JAX-based library for building and running neural networks with ease.
- stars
- 2.2K
- forks
- 144
- watching
- 25
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
elixir-nx/axon1.6K
Neural Network Framework
Provides a framework for building and training neural networks using Elixir and Nx.
aria42/flare285
Neural graph library
A Clojure library for building and manipulating dynamic neural net graphs with pluggable tensor operations and hardware acceleration.
Differential equation solver
Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX.
Network builder
Builds neural networks with less boilerplate code by providing a standardized interface for different architectures
Neural net library
A Clojure-based library for building and training neural networks
Neural network builder
A deep learning framework on top of PyTorch for building neural networks.
Graph library
A lightweight library for working with graph neural networks in jax.
SNN library
A JAX-based library for training and utilizing spiking neural networks
Neural computer framework
An implementation of Neural Turing Machine and Differentiable Neural Computer architectures using PyTorch and Visdom for deep learning tasks.
cloud-cv/fabrik1.1K
Neural net builder
An online platform to design and visualize neural networks collaboratively
gigasquid/k9104
Neural network library
A Clojure library for constructing and training simple neural networks using core.matrix
norse/norse685
SNN framework
Deep learning framework for spiking neural networks
Training framework
A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments.
Neural network library
A Clojure wrapper around the Encog machine learning framework, providing an interface to build and train neural networks
Neural computer
An implementation of the Differentiable Neural Computer architecture in PyTorch
Featured in 1 awesome list
Each link jumps to the spot where the list mentions equinox.
