Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for neural network jax
jax
x
neural-network
x
16 search results found
Keras
⭐
61,137
Deep Learning for humans
Ivy
⭐
13,904
The Unified AI Framework
Dm Haiku
⭐
2,807
JAX-based neural network library
Pennylane
⭐
2,166
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
Neural Tangents
⭐
2,149
Fast and Easy Infinite Neural Networks in Python
Deepxde
⭐
2,135
A library for scientific machine learning and physics-informed learning
Equinox
⭐
1,614
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Machine_learning_refined
⭐
1,574
Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).
Awesome Jax
⭐
1,156
JAX - A curated list of resources https://github.com/google/jax
Jraph
⭐
1,141
A Graph Neural Network Library in Jax
Diffrax
⭐
1,131
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Fortuna
⭐
824
A Library for Uncertainty Quantification.
Pdebench
⭐
522
PDEBench: An Extensive Benchmark for Scientific Machine Learning
Netket
⭐
476
Machine learning algorithms for many-body quantum systems
Elegy
⭐
430
A High Level API for Deep Learning in JAX
Git Re Basin
⭐
378
Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"
Learning Python Physics Informed Machine Learning Pinns Deeponets
⭐
297
Physics Informed Machine Learning Tutorials (Pytorch and Jax)
Tensorcircuit
⭐
230
Tensor network based quantum software framework for the NISQ era
Optimistix
⭐
185
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Jaxnet
⭐
171
Concise deep learning for JAX
Sunstreaker
⭐
126
以jax为后端的类似keras的框架
Fbpinns
⭐
110
Solve forward and inverse problems related to partial differential equations using finite basis physics-informed neural networks (FBPINNs)
Ngc Learn
⭐
71
NGC-Learn: Predictive Coding and Neurobiologically-Motivated Learning in Python
Nnx
⭐
70
Neural Networks for JAX
Jax Resnet
⭐
67
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Hyper Nn
⭐
65
Easy Hypernetworks in Pytorch and Jax
Dks
⭐
50
Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
Madam
⭐
45
Pytorch and Jax code for the Madam optimiser.
Spyx
⭐
31
Spyx: Spiking Neural Networks in JAX
Pyhgf
⭐
27
PyHGF: A graph neural network library for predictive coding
Inox
⭐
21
Stainless neural networks in JAX
Connex
⭐
19
Fine-grained, dynamic control of neural network topology in JAX.
Scimlbenchmarksoutput
⭐
13
SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance
Deep Learning Projects
⭐
13
Best Deep Learning Projects for Advanced Learners
Veni
⭐
11
A simple Python package for deep learning using forward automatic differentiation based on JAX.
Energy Transformer Graph
⭐
11
This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification
Yerbamate
⭐
10
A framework-agnostic deep learning package and experiment manager
Doc Builder
⭐
6
Simple Documentation Builder for Ivy Projects.
Jax Nn Template
⭐
5
Deep Learning Template with JAX
Netket_fidelity
⭐
5
Projected time-dependent Variational Monte Carlo (p-tVMC) method based on infidelity optimization for variational simulation of quantum dynamics.
Related Searches
Python Neural Network (7,064)
Machine Learning Neural Network (4,430)
Deep Learning Neural Network (3,760)
Jupyter Notebook Neural Network (3,677)
Tensorflow Neural Network (2,192)
Pytorch Neural Network (1,416)
Keras Neural Network (1,309)
Neural Network Classification (906)
C Plus Plus Neural Network (856)
Neural Network Artificial Intelligence (829)
1-16 of 16 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.