The Wayback Machine - https://web.archive.org/web/20200820061942/https://github.com/topics/jax

#

jax

Here are 50 public repositories matching this topic...

google / trax

Star

Trax — Deep Learning with Clear Code and Speed

machine-learning reinforcement-learning deep-learning numpy deep-reinforcement-learning transformer jax

Updated Aug 20, 2020
Python

explosion / thinc

Star

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

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 Aug 19, 2020
Python

bethgelab / foolbox

Star

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

python machine-learning tensorflow keras pytorch adversarial-examples adversarial-attacks jax

Updated Aug 19, 2020
Python

google / neural-tangents

Star

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 Aug 16, 2020
Jupyter Notebook

google-research / big_transfer

Star

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

deep-learning pytorch imagenet convolutional-neural-networks transfer-learning jax tensorflow2

Updated Aug 19, 2020
Python

deepmind / dm-haiku

Star

JAX-based neural network library

machine-learning deep-neural-networks deep-learning neural-networks jax

Updated Aug 14, 2020
Python

pyro-ppl / numpyro

Star

Open

[FR] SinhArcsinh transformation

fehiepsi commented Aug 14, 2020

Requested by @nkaimcaudle at pyro-ppl/numpyro#703 (comment).

Read more

good first issue

Open

Integrate with flax/haiku

4

Open

Implement some popular distributions

4

Find more good first issues →

jonasrauber / eagerpy

Star

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code

python tensorflow numpy pytorch jax eager-execution tensorflow2

Updated Aug 14, 2020
Python

JuliusKunze / jaxnet

Star

Alternative to TensorFlow2/Keras/PyTorch for more concise, robust and optimized deep learning code

python data-science machine-learning deep-learning neural-networks jax

Updated Aug 18, 2020
Python

google-research / fast-soft-sort

Star

Fast Differentiable Sorting and Ranking

sorting tensorflow pytorch ranking differentiable jax

Updated Jul 28, 2020
Python

pyhf

scikit-hep / pyhf

Star

Open

change to standard NLL fit objective

3

lukasheinrich commented Jul 29, 2020

right now we use "twice_nll" as a fit objective and in the test statistic a simple diffence

twice_nll_constrfit - twice_nll_globalfit

but rather we should just to a NLL fit and in the test stat do

2*(nll_constrfit - nll_globalfit)

this will require updating some test reference numbers in the tests

Read more

good first issue help wanted

Open

Loading json schema via pkgutil

Open

Check for extraneous tensor casts with "data" handling

Find more good first issues →

dionhaefner / pyhpc-benchmarks

Star

A suite of benchmarks to test the sequential CPU and GPU performance of most popular high-performance libraries for Python.

python tensorflow gpu parallel-computing pytorch high-performance-computing benchmarks cupy jax

Updated Aug 3, 2020
Python

SymJAX / SymJAX

Star

Documentation:

deep-neural-networks lasagne theano deep-learning tensorflow numpy dataset jax

Updated Aug 19, 2020
Python

easy-neural-ode

jacobjinkelly / easy-neural-ode

Star

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 Jul 18, 2020
Python

oughtinc / ergo

Star

Open

Remove unnecessary redirect in Metaculus API requests

stuhlmueller commented Jun 7, 2020

Because / at end of question is missing

Read more

good first issue

Open

Update all notebooks to work with latest Ergo

Open

Add tests to improve code coverage

jxbz / madam

Star

Pytorch and Jax code for the Madam optimiser.

deep-learning optimizer pytorch neural-networks jax madam

Updated Jul 19, 2020
Jupyter Notebook

IvanYashchuk / jax-fenics-adjoint

Star

Differentiable interface to FEniCS for JAX using dolfin-adjoint/pyadjoint

automatic-differentiation fenics adjoint adjoint-sensitivities jax adjoint-solvers

Updated Aug 14, 2020
Jupyter Notebook

dnet

umangjpatel / dnet

Star

Open

Documentation and experiments

umangjpatel commented Jan 17, 2020

Description
Update README and add experiments.

Solution
Create experiments or examples package to work

Read more

documentation good first issue

Rahlir / OmniFocusScripts

Star

Collection of useful omnifocus applescripts

productivity automation applescript extensions icons omnifocus omni jax omnifocus-library omnifocus3 omnigroup

Updated Jan 24, 2019
AppleScript

IvanYashchuk / jax-fenics

Star

Differentiable interface to FEniCS for JAX

automatic-differentiation fenics adjoint jax

Updated Apr 17, 2020
Python

henry-prior / jax-rl

Star

JAX implementations of core Deep RL algorithms

reinforcement-learning deep-learning deep-reinforcement-learning flax deepmind sac actor-critic maximum-a-posteriori-estimation mujoco jax td3 soft-actor-critic

Updated Apr 21, 2020
Python

MinRegret / TigerControl

Star

Google AI Princeton control framework

python benchmarking machine-learning control autograd pybullet jax

Updated Aug 12, 2020
Jupyter Notebook

tesslerc / TD3-JAX

Star

A JAX Implementation of the Twin Delayed DDPG Algorithm

reinforcement-learning deep-reinforcement-learning mujoco jax

Updated Mar 12, 2020
Jupyter Notebook

gcucurull / jax-gat

Star

JAX implementation of Graph Attention Networks

neural-networks attention-mechanism graph-attention-networks jax graph-neural-networks

Updated Apr 26, 2020
Python

gcucurull / jax-gcn

Star

Graph Convolutional Networks in JAX

graph-convolutional-networks gcn jax graph-neural-networks

Updated May 8, 2020
Python

PhilipVinc / mpi4jax

Star

MPI greets Jax and speeds it ups

mpi jit jax xla

Updated Aug 18, 2020
Python

berenslab / RFEst

Star

A Python 3 toolbox for neural receptive field estimation using splines and Gaussian priors.

matrix-factorization splines jax receptive-field evidence-optimization

Updated Aug 19, 2020
Jupyter Notebook

niklasschmitz / DeepQLearning

Star

small experiments with agents learning atari games, implemented in jax/numpy

deep-reinforcement-learning deep-q-learning jax

Updated Mar 16, 2019
Python

shekohex / jax

Star

minimal C-interpreter to play with. for learning purpose

c functional-programming scripting-language learning-by-doing jax

Updated Jan 9, 2018
C

vicgalle / sgmcmc-force

Star

Samplers from the paper "Stochastic Gradient MCMC with Repulsive Forces"

bayesian-inference mcmc-sampler jax svgd sgld sgmcmc

Updated Feb 3, 2020
Jupyter Notebook

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."

You can’t perform that action at this time.