neural-tangents
Neural Network Library
A high-level neural network API for defining and training complex hierarchical networks of finite or infinite width
Fast and Easy Infinite Neural Networks in Python
2k stars
62 watching
227 forks
Language: Jupyter Notebook
last commit: over 2 years agoLinked from 2 awesome lists
bayesian-inferencebayesian-networksdeep-networksgaussian-processesgradient-descentgradient-flowinfinite-networksjaxkernelkernel-computationneural-networksneural-tangentstraining-dynamics
n2cholas/awesome-jax
amrzv/awesome-colab-notebooksRelated projects:
| Repository | Description | Stars |
|---|---|---|
 nixtla/neuralforecast | An open-source Python library providing a unified interface to various neural forecasting models, focusing on performance, usability, and robustness. | 3,181 |
 karpathy/neuraltalk | A system for generating image descriptions using neural networks | 5,414 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
 lucidrains/imagen-pytorch | Implements Google's Text-to-Image Neural Network in PyTorch using a cascading DDPM architecture with dynamic clipping and noise level conditioning. | 8,127 |
 guillaume-chevalier/hyperopt-keras-cnn-cifar-100 | Automates hyperparameter optimization and neural network architecture search using Hyperopt on a CNN model for the CIFAR-100 dataset | 106 |
 jax-ml/jax | A library that provides high-performance numerical computing and machine learning capabilities. | 30,744 |
| google/flax | Provides a flexible neural network library and ecosystem for JAX | 6,196 |
 benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,694 |
 nvlabs/instant-ngp | A software toolkit for training and rendering neural graphics primitives | 16,115 |
 leriomaggio/deep-learning-keras-tensorflow | A comprehensive tutorial on building and training deep neural networks using Keras and TensorFlow | 2,951 |
 trekhleb/homemade-machine-learning | Practices implementing popular machine learning algorithms from scratch to gain a deeper understanding of their mathematics | 23,191 |
 sony/nnabla | A deep learning framework that provides a flexible and expressive Python API for building and training neural networks on various platforms. | 2,729 |
 minimaxir/textgenrnn | A Python module for creating character-level or word-level neural networks for text generation and training on various datasets | 4,944 |
 neuralhydrology/neuralhydrology | A Python library for training neural networks with focus on hydrological applications using PyTorch. | 372 |
| benedekrozemberczki/capsgnn | A PyTorch implementation of a graph neural network architecture | 1,246 |