#
jax
Here are 131 public repositories matching this topic...
Python code for "Machine learning: a probabilistic perspective" (2nd edition)
-
Updated
Apr 24, 2021 - Jupyter Notebook
Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
-
Updated
Apr 3, 2021 - Python
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
-
Updated
Apr 23, 2021 - Python
python
nlp
machine-learning
natural-language-processing
ai
deep-learning
mxnet
functional-programming
tensorflow
pytorch
artificial-intelligence
spacy
machine-learning-library
type-checking
jax
-
Updated
Apr 21, 2021 - Python
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
-
Updated
Apr 20, 2021 - Python
Flax is a neural network library for JAX that is designed for flexibility.
-
Updated
Apr 24, 2021 - Python
Fast and Easy Infinite Neural Networks in Python
kernel
neural-networks
gradient-descent
bayesian-inference
gaussian-processes
bayesian-networks
deep-networks
gradient-flow
jax
infinite-networks
training-dynamics
neural-tangents
kernel-computation
-
Updated
Apr 21, 2021 - Jupyter Notebook
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
-
Updated
Feb 9, 2021 - Python
JAX-based neural network library
-
Updated
Apr 24, 2021 - Python
TensorLy: Tensor Learning in Python.
python
machine-learning
mxnet
tensorflow
numpy
pytorch
decomposition
tensor-factorization
tensor
tensor-algebra
tensorly
tensor-learning
tensor-decomposition
cupy
tensor-regressions
tensor-methods
jax
-
Updated
Apr 24, 2021 - Python
fehiepsi
commented
Apr 8, 2021
Currently, in AutoLaplaceApproximation, we use jax.hessian to compute Hessian matrix. But the hessian matrix is not always positive definite. So it would be nice to use Gaussian-Newton hessian matrix when scale_tril = Nan.
A Graph Neural Network Library in Jax
-
Updated
Mar 17, 2021 - Python
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
-
Updated
Apr 23, 2021 - Python
Fast Differentiable Sorting and Ranking
-
Updated
Jul 28, 2020 - Python
Long Range Arena for Benchmarking Efficient Transformers
-
Updated
Apr 13, 2021 - Python
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
flax
stochastic-differential-equations
inverse-problems
generative-models
jax
score-matching
iclr2021
score-based-generative-modeling
controllable-generation
-
Updated
Apr 8, 2021 - Jupyter Notebook
JAX - A curated list of resources https://github.com/google/jax
-
Updated
Mar 22, 2021
Elegy is a framework-agnostic Trainer interface for the Jax ecosystem.
-
Updated
Apr 2, 2021 - Python
matthewfeickert
commented
Apr 15, 2021
Description
At the PyHEP April 2021 topical meeting pyhf tutorial, @BlaiseDelaney had the following question on pertaining to the Normalisation Uncertainty (normsys) docs:
I'm a bit confused about what I need to supply to the modifier. Is it the actual nume
Concise deep learning for JAX
-
Updated
Oct 21, 2020 - Python
fritzo
commented
Mar 29, 2021
See wikipedia table for example patterns: https://en.wikipedia.org/wiki/List_of_integrals_of_Gaussian_functions
Code for the paper "Learning Differential Equations that are Easy to Solve"
machine-learning
deep-neural-networks
deep-learning
ode
dynamical-systems
differential-equations
numerical-integration
ode-solver
jax
neural-ode
neural-differential-equations
-
Updated
Dec 15, 2020 - Python
The templated deep learning framework, enabling framework-agnostic functions, layers and libraries.
python
template
machine-learning
deep-learning
neural-network
mxnet
tensorflow
gpu
numpy
pytorch
autograd
abstraction
ivy
jax
-
Updated
Apr 22, 2021 - Python
A suite of benchmarks to test the sequential CPU and GPU performance of most popular high-performance libraries for Python.
-
Updated
Apr 5, 2021 - Python
Jax (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
reinforcement-learning
deep-learning
deep-reinforcement-learning
gym
flax
behavioral-cloning
sac
continuous-control
deep-deterministic-policy-gradient
jax
soft-actor-critic
offline-reinforcement-learning
batch-reinforcement-learning
-
Updated
Apr 24, 2021 - Jupyter Notebook
Documentation:
-
Updated
Dec 21, 2020 - Python
A Python library for integrating model-based and judgmental forecasting
-
Updated
Mar 25, 2021 - Jupyter Notebook
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
-
Updated
Apr 21, 2021 - Python
Improve this page
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."
Is there a way to train a bidirectional RNN (like LSTM or GRU) on trax nowadays?