vnn-jax-enn

Variational Neural Networks JAX implementation

Stars

2

Forks

1

Language

Python

Last Updated

Jul 05, 2022

Similar Repos
Repo Language Stars Description Updated At
Python 2 Variational Neural Networks Pytoch and JAX implementations Jan 10, 2023
None 2 Variational Neural Networks Pytoch and JAX implementations May 19, 2023
Python 4 Variational Neural Networks Pytorch implementation Mar 11, 2023
Python 2 Jax/Flax implementation practice for various neural networks Mar 22, 2023
Python 38 Jax/Flax implementation of Variational-DiffWave. Apr 16, 2023
Python 11 Stainless neural networks in JAX Nov 15, 2023
Python 96 Variational Graph Recurrent Neural Networks - PyTorch May 02, 2023
Python 2 Neural Networks based Variational Mode Decomposition Mar 21, 2023
Python 7 An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN) Jul 07, 2022
Python 11 JAX implementation of Graph Attention Networks Jul 13, 2022
Jupyter Notebook 5 A short tutorial on Bayesian Neural Networks with JAX Mar 03, 2023
Python 21 A collection of graph neural networks implementations in JAX May 09, 2023
Python 28 Build neural networks for machine learning force fields with JAX May 01, 2023
Jupyter Notebook 15 JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et … Apr 14, 2023
Python 73 Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779) Oct 08, 2022
Python 30 🎬🧠 Exploring neural networks (and variational inference) for collaborative filtering May 22, 2022
Python 6 synthetic data experiments for over-pruning in variational bayesian neural networks Feb 17, 2021
None 2 Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX. Mar 12, 2023
Python 1098 Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/ Apr 27, 2023
D 2 neural networks implementation in D Dec 12, 2012
Haskell 2 Haskell implementation of Neural Networks Sep 26, 2022
Jupyter Notebook 25 Implementation of several Generative Adversarial Networks in JAX / Flax Apr 23, 2023
Python 541 TensorFlow implementation of Neural Variational Inference for Text Processing Jul 26, 2022
Jupyter Notebook 88 Sample code for running deterministic variational inference to train Bayesian neural networks Jul 13, 2022
Python 19 Variational Dropout Sparsifies Deep Neural Networks (Molchanov et al. 2017) by Chainer Dec 04, 2022
Python 2 We use Variational Artificial Neural Networks to solve the nuclear bound state Jul 29, 2023
Jupyter Notebook 2 Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax. Feb 08, 2024
JavaScript 2 artificial neural networks implementation using SGD Jan 24, 2020
Jupyter Notebook 3 Convolution Neural Networks implementation in TensorFlow Feb 19, 2023
Jupyter Notebook 22 A variational implementation of classical and quantum annealing using recurrent neural networks for the purpose … Oct 06, 2022
Python 13 Meta-learning synaptic plasticity in deep neural networks with connectomic constraints in JAX Feb 17, 2023
Python 29 Graph Convolutional Networks in JAX May 28, 2023
Python 4 Variational autoencoders on categorical and continuous data using jax Feb 12, 2023
None 4 Implementation of Simple Neural Networks with Python Jul 22, 2021
Python 7 PyTorch implementation of Hamiltonian deep neural networks. May 13, 2022
Python 2 Implementation of a Feed Forward Neural Networks. Dec 16, 2021
Python 2 Tensorflow implementation fo Quasi-Recurrent Neural Networks Mar 28, 2020
Python 2 Code implementation of major Convolutional Neural Networks Feb 01, 2020
Python 85 Implementation of Artificial Neural Networks using NumPy Jul 06, 2022
Python 290 Keras implementation of Non-local Neural Networks Apr 21, 2023
Python 236 Clean implementation of feed forward neural networks Apr 14, 2023
Python 4 dm-haiku implementation of hyperbolic neural networks Apr 14, 2023
Python 2 TensorFlow implementation of Pixel Recurrent Neural Networks Feb 10, 2018
Jupyter Notebook 3 Implementation of Logic Gates using Neural Networks Mar 31, 2023
Python 18 pytorch implementation of DAG-Recurrent Neural Networks Mar 19, 2023
Python 47 Official PyTorch implementation of Hyperbolic Neural Networks++ Apr 26, 2023
Python 100 Complex-valued neural networks for pytorch and Variational Dropout for real and complex layers. Sep 09, 2022
Python 2095 JAX-based neural network library Aug 15, 2022
Python 107 Neural network verification in JAX Aug 12, 2022
None 7 JAX-based neural network library Jul 30, 2020