Here are
47 public repositories
matching this topic...
Trax — Deep Learning with Clear Code and Speed
-
Updated
Jul 31, 2020
-
Python
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
-
Updated
Jul 31, 2020
-
Python
Fast and Easy Infinite Neural Networks in Python
-
Updated
Jul 27, 2020
-
Jupyter Notebook
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
-
Updated
Jul 20, 2020
-
Python
JAX-based neural network library
-
Updated
Jul 29, 2020
-
Python
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
-
Updated
Jul 27, 2020
-
Python
Alternative to TensorFlow2/Keras/PyTorch for more concise, robust and optimized deep learning code
-
Updated
May 17, 2020
-
Python
Fast Differentiable Sorting and Ranking
-
Updated
Jul 28, 2020
-
Python
pure-python implementation of some (maybe someday all?) HistFactory models
-
Updated
Jul 31, 2020
-
Python
A suite of benchmarks to test the sequential CPU and GPU performance of most popular high-performance libraries for Python.
-
Updated
Jul 22, 2020
-
Python
-
Updated
Jul 27, 2020
-
Python
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
-
Updated
Jul 15, 2020
-
Python
Code for the paper "Learning Differential Equations that are Easy to Solve"
-
Updated
Jul 18, 2020
-
Python
A Python library for integrating model-based and judgmental forecasting
-
Updated
Jul 29, 2020
-
Jupyter Notebook
Pytorch and Jax code for the Madam optimiser.
-
Updated
Jul 19, 2020
-
Jupyter Notebook
Differentiable interface to FEniCS for JAX using dolfin-adjoint/pyadjoint
-
Updated
Jul 17, 2020
-
Jupyter Notebook
-
Updated
May 15, 2020
-
Python
Collection of useful omnifocus applescripts
-
Updated
Jan 24, 2019
-
AppleScript
Differentiable interface to FEniCS for JAX
-
Updated
Apr 17, 2020
-
Python
Google AI Princeton control framework
-
Updated
Apr 1, 2020
-
Jupyter Notebook
JAX implementations of core Deep RL algorithms
-
Updated
Apr 21, 2020
-
Python
A JAX Implementation of the Twin Delayed DDPG Algorithm
-
Updated
Mar 12, 2020
-
Jupyter Notebook
JAX implementation of Graph Attention Networks
-
Updated
Apr 26, 2020
-
Python
Graph Convolutional Networks in JAX
-
Updated
May 8, 2020
-
Python
MPI greets Jax and speeds it ups
-
Updated
Jul 23, 2020
-
Python
A Python 3 toolbox for neural receptive field estimation using splines and Gaussian priors.
-
Updated
Jul 28, 2020
-
Jupyter Notebook
small experiments with agents learning atari games, implemented in jax/numpy
-
Updated
Mar 16, 2019
-
Python
minimal C-interpreter to play with. for learning purpose
Samplers from the paper "Stochastic Gradient MCMC with Repulsive Forces"
-
Updated
Feb 3, 2020
-
Jupyter Notebook
-
Updated
Jul 13, 2020
-
Shell
Improve this page
Add a description, image, and links to the
jax
topic page so that developers can more easily learn about it.
Curate this topic
Add this topic to your repo
To associate your repository with the
jax
topic, visit your repo's landing page and select "manage topics."
Learn more
You can’t perform that action at this time.
You signed in with another tab or window. Reload to refresh your session.
You signed out in another tab or window. Reload to refresh your session.