jax

Star

Here are 495 public repositories matching this topic...

huggingface / transformers

Star

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

Updated Dec 21, 2022
Python

google / jax

Star

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

jax

Updated Dec 21, 2022
Python

d2l-ai / d2l-en

Star

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 400 universities from 60 countries including Stanford, MIT, Harvard, and Cambridge.

python data-science machine-learning natural-language-processing reinforcement-learning computer-vision deep-learning mxnet book notebook tensorflow keras pytorch kaggle hyperparameter-optimization recommender-system gaussian-processes jax

Updated Dec 20, 2022
Python

ml-tooling / best-of-ml-python

Star

🏆 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 Dec 17, 2022

unifyai / ivy

Star

The Unified Machine Learning Framework

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

Updated Dec 21, 2022
Python

google / trax

Star

Trax — Deep Learning with Clear Code and Speed

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

Updated Dec 19, 2022
Python

arogozhnikov / einops

Star

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

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

Updated Dec 18, 2022
Python

probml / pyprobml

Star

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

BoltzmannEntropy / interviews.ai

Star

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

Star

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

jax

Updated Dec 20, 2022
Python

tensorflow / datasets

Star

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

data machine-learning tensorflow numpy dataset datasets jax

Updated Dec 21, 2022
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 Dec 20, 2022
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 Nov 22, 2022
Python

deepmind / dm-haiku

Star

JAX-based neural network library

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

Updated Dec 9, 2022
Python

grame-cncm / faust

Star

Functional programming language for signal processing and sound synthesis

audio c rust c-plus-plus csharp compiler cpp functional-programming julia dsp llvm dlang julia-language wasm faust jax cmajor

Updated Dec 20, 2022
C++

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

PennyLaneAI / pennylane

Star

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