inox

Stainless neural networks in JAX

Stars

23

Forks

0

Language

Python

Last Updated

Feb 13, 2024

Similar Repos
Repo Language Stars Description Updated At
Python 2 Variational Neural Networks JAX implementation Jul 05, 2022
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 2 Jax/Flax implementation practice for various neural networks Mar 22, 2023
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
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
Python 7 An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN) Jul 07, 2022
Jupyter Notebook 2 Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax. Feb 08, 2024
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 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
None 2 JAX-based neural network library Feb 01, 2024
Python 11 JAX implementation of Graph Attention Networks Jul 13, 2022
Python 3 Exploring connections between transformer neural networks, spin models, and (non-)equilibrium statistical mechanics in JAX. Apr 24, 2023
Jupyter Notebook 3 Neural Cellular Automata implemented with Jax Aug 12, 2022
Jupyter Notebook 2 Neural networks. Mar 24, 2022
Python 3 Neural Networks Nov 26, 2020
MATLAB 5 Neural Networks Dec 22, 2020
Jupyter Notebook 15 JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et … Apr 14, 2023
Python 2486 A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and … May 21, 2023
Python 3 Collection of templates and examples for implementing spiking neural networks in plain PyTorch and JAX … May 05, 2023
Jupyter Notebook 2 Basics of neural networks, implementing neural networks from scratch May 26, 2022
Python 991 A Graph Neural Network Library in Jax Aug 15, 2022
C 5 Convolutional Neural Networks Jul 28, 2022
Python 298 Lagrangian Neural Networks Aug 15, 2022
None 2 Convolutional Neural Networks Feb 18, 2022
None 2 Convolutional Neural Networks Nov 23, 2021
C 2 Convolutional Neural Networks Apr 23, 2020
C 3 Convolutional Neural Networks May 17, 2021
C 3 Convolutional Neural Networks Jun 29, 2022
Python 33 spiking-neural-networks Aug 18, 2022
JavaScript 2 javascript neural networks Aug 14, 2013
None 16 Convolutional Neural Networks Aug 02, 2022
C 23371 Convolutional Neural Networks Oct 17, 2022
None 2 Convolutional Neural Networks Feb 17, 2023
Python 16 Kervolutional neural networks Aug 28, 2021
JavaScript 2 Learning Neural Networks Aug 18, 2015
C 2 Convolutional Neural Networks Sep 06, 2017
Go 16 Genetic Neural Networks Feb 24, 2023
Python 39 Bayesian Neural Networks Nov 16, 2022
C 3 Convolutional Neural Networks Jun 28, 2022
None 6 Convolutional Neural Networks Mar 04, 2023
C 2 Neural Networks i.MX Mar 12, 2023
C 6 Convolutional Neural Networks Jul 06, 2022
C 2 Convolutional Neural Networks Feb 28, 2022