equinox
Neural network library
A JAX-based library for building and running neural networks with ease.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2k stars
25 watching
144 forks
Language: Python
last commit: 3 months ago
Linked from 1 awesome list
deep-learningequinoxjaxneural-networks
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 elixir-nx/axon | Provides a framework for building and training neural networks using Elixir and Nx. | 1,562 |
 aria42/flare | A Clojure library for building and manipulating dynamic neural net graphs with pluggable tensor operations and hardware acceleration. | 285 |
| patrick-kidger/diffrax | Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX. | 1,480 |
 p-christ/nn_builder | Builds neural networks with less boilerplate code by providing a standardized interface for different architectures | 166 |
 japonophile/synaptic | A Clojure-based library for building and training neural networks | 88 |
 ramon-oliveira/aorun | A deep learning framework on top of PyTorch for building neural networks. | 61 |
 google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
 kmheckel/spyx | A JAX-based library for training and utilizing spiking neural networks | 104 |
 jingweiz/pytorch-dnc | An implementation of Neural Turing Machine and Differentiable Neural Computer architectures using PyTorch and Visdom for deep learning tasks. | 278 |
 cloud-cv/fabrik | An online platform to design and visualize neural networks collaboratively | 1,127 |
 gigasquid/k9 | A Clojure library for constructing and training simple neural networks using core.matrix | 104 |
 norse/norse | Deep learning framework for spiking neural networks | 685 |
 erotemic/netharn | A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. | 40 |
 jimpil/enclog | A Clojure wrapper around the Encog machine learning framework, providing an interface to build and train neural networks | 137 |
 ypxie/pytorch-neucom | An implementation of the Differentiable Neural Computer architecture in PyTorch | 94 |