#

jax

Here are 405 public repositories matching this topic...

transformers

huggingface / transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Updated Aug 31, 2022
Python

jax

google / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

jax

Updated Aug 31, 2022
Python

ml-tooling / best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

python nlp data-science machine-learning awesome deep-learning tensorflow scikit-learn python-library keras ml data-visualization pytorch transformer data-analysis automl jax data-visualizations best-of

Updated Aug 29, 2022

google / trax

Trax — Deep Learning with Clear Code and Speed

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

Updated Aug 8, 2022
Python

einops

arogozhnikov / einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

deep-learning chainer tensorflow numpy keras pytorch tensor gluon cupy jax

Updated Aug 24, 2022
Python

unifyai / ivy

The Unified Machine Learning Framework

python template machine-learning deep-learning neural-network mxnet tensorflow gpu numpy pytorch autograd abstraction ivy jax

Updated Aug 31, 2022
Python

probml / pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy

machine-learning tensorflow pytorch colab pml probabilistic-programming flax jupyter-notebooks pymc3 pyro jax numpyro blackjax

Updated Aug 28, 2022
Jupyter Notebook

BoltzmannEntropy / interviews.ai

It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.

python data-science machine-learning deep-learning information-theory jobs pytorch autograd artificial-intelligence feature-extraction ensemble-learning logistic-regression convolutional-neural-networks bayesian-statistics loss-functions interview-preparation pytorch-tutorial graduate-school jax

Updated Jan 21, 2022

google / flax

Flax is a neural network library for JAX that is designed for flexibility.

jax

Updated Aug 31, 2022
Python

tensorflow / datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

data machine-learning tensorflow numpy dataset datasets jax

Updated Aug 31, 2022
Python

explosion / thinc

🔮 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 29, 2022
Python

bethgelab / foolbox

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 Jul 29, 2022
Python

deepmind / dm-haiku

JAX-based neural network library

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

Updated Aug 30, 2022
Python

google / neural-tangents

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 19, 2022
Jupyter Notebook

pennylane

PennyLaneAI / pennylane

PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.

python machine-learning deep-learning neural-network qml tensorflow optimization quantum differentiable-computing automatic-differentiation pytorch autograd quantum-computing quantum-chemistry quantum-machine-learning qiskit jax cirq strawberryfields

Updated Aug 31, 2022
Python

pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

numpy probabilistic-programming bayesian-inference mcmc hmc pyro jax inference-algorithms

Updated Aug 29, 2022
Python

google / brax

Massively parallel rigidbody physics simulation on accelerator hardware.

reinforcement-learning robotics physics-simulation jax

Updated Aug 18, 2022
Jupyter Notebook

google-research / big_transfer

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

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

Updated Jun 22, 2022
Python

google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

research computer-vision deep-learning transformers attention jax vision-transformer

Updated Aug 31, 2022
Python

tensorly

tensorly / tensorly

TensorLy: Tensor Learning in Python.

python machine-learning mxnet tensorflow numpy regression pytorch decomposition tensor-factorization tensor tensor-algebra tensorly tensor-learning tensor-decomposition cupy tensor-methods jax tensor-regression

Updated Aug 29, 2022
Python

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