Topic: jax Goto Github

👇 Here are 919 public repositories matching this topic...

alpa-projects / alpa
jax,Training and serving large-scale neural networks with auto parallelization.
Organization: alpa-projects
Home Page: https://alpa.ai
deep-learning machine-learning compiler distributed-training high-performance-computing alpa jax distributed-computing llm auto-parallelization

apple / axlearn
jax,An Extensible Deep Learning Library
Organization: apple
deep-learning jax

arogozhnikov / einops
jax,Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
User: arogozhnikov
Home Page: https://einops.rocks
chainer cupy deep-learning einops jax keras numpy pytorch tensor tensorflow

ashishpatel26 / treasure-of-transformers
jax,💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️
User: ashishpatel26
Home Page: https://github.com/ashishpatel26/Treasure-of-Transformers.git
awesome bert jax language-model language-models model-hub natural-language-generation natural-language-processing natural-language-understanding nlp nlp-library pretrained-models python pytorch pytorch-transformers seq2seq speech-recognition tensorflow transformer

awslabs / fortuna
jax,A Library for Uncertainty Quantification.
Organization: awslabs
Home Page: https://aws-fortuna.readthedocs.io/en/latest/
deep-learning neural-networks uncertainty uncertainty-estimation uncertainty-quantification bayesian-inference calibration conformal-prediction model-calibration uncertainty-calibration

bethgelab / foolbox
jax,A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Organization: bethgelab
Home Page: https://foolbox.jonasrauber.de
adversarial-examples machine-learning python adversarial-attacks pytorch tensorflow jax keras

boltzmannentropy / interviews.ai
jax,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.
User: boltzmannentropy
data-science machine-learning deep-learning interview-preparation jobs artificial-intelligence pytorch-tutorial graduate-school pytorch python

d2l-ai / d2l-en
jax,Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Organization: d2l-ai
Home Page: https://D2L.ai
deep-learning machine-learning book notebook computer-vision natural-language-processing python kaggle data-science mxnet

danijar / dreamerv3
jax,Mastering Diverse Domains through World Models
User: danijar
Home Page: https://danijar.com/dreamerv3
artificial-intelligence general minecraft reinforcement-learning world-models jax

dbraun / dawdreamer
jax,Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX, Warp Markers, and JUCE processors
User: dbraun
juce audio python daw vst audio-plugin audio-processing faust vst3 vst3-host

explosion / thinc
jax,🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Organization: explosion
Home Page: https://thinc.ai
machine-learning deep-learning artificial-intelligence ai python nlp natural-language-processing spacy machine-learning-library type-checking

google-deepmind / dm-haiku
jax,JAX-based neural network library
Organization: google-deepmind
Home Page: https://dm-haiku.readthedocs.io
machine-learning neural-networks jax deep-learning deep-neural-networks

google-deepmind / jraph
jax,A Graph Neural Network Library in Jax
Organization: google-deepmind
Home Page: https://jraph.readthedocs.io/en/latest/
graph-neural-networks jax machine-learning deep-learning

google-deepmind / mctx
jax,Monte Carlo tree search in JAX
Organization: google-deepmind
jax reinforcement-learning monte-carlo-tree-search

google-deepmind / penzai
jax,A JAX research toolkit for building, editing, and visualizing neural networks.
Organization: google-deepmind
Home Page: https://penzai.readthedocs.io/
fine-tuning interpretability jax neural-networks visualization

google-research / big_transfer
jax,Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Organization: google-research
Home Page: https://arxiv.org/abs/1912.11370
deep-learning convolutional-neural-networks imagenet tensorflow2 jax pytorch transfer-learning

google-research / scenic
jax,Scenic: A Jax Library for Computer Vision Research and Beyond
Organization: google-research
jax computer-vision deep-learning research attention transformers vision-transformer

google / brax
jax,Massively parallel rigidbody physics simulation on accelerator hardware.
Organization: google
jax reinforcement-learning robotics physics-simulation

google / flax
jax,Flax is a neural network library for JAX that is designed for flexibility.
Organization: google
Home Page: https://flax.readthedocs.io
jax

google / jax
jax,Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Organization: google
Home Page: http://jax.readthedocs.io/
jax

google / jaxopt
jax,Hardware accelerated, batchable and differentiable optimizers in JAX.
Organization: google
Home Page: https://jaxopt.github.io
jax optimization deep-learning differentiable-programming bi-level

google / neural-tangents
jax,Fast and Easy Infinite Neural Networks in Python
Organization: google
Home Page: https://iclr.cc/virtual_2020/poster_SklD9yrFPS.html
neural-networks infinite-networks gaussian-processes training-dynamics neural-tangents jax kernel deep-networks kernel-computation gradient-descent

google / trax
jax,Trax — Deep Learning with Clear Code and Speed
Organization: google
jax numpy deep-learning deep-reinforcement-learning machine-learning transformer reinforcement-learning

grame-cncm / faust
jax,Functional programming language for signal processing and sound synthesis
Organization: grame-cncm
Home Page: http://faust.grame.fr
faust dsp audio compiler wasm c cpp llvm functional-programming rust

huggingface / diffusers
jax,🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Organization: huggingface
Home Page: https://huggingface.co/docs/diffusers
deep-learning diffusion image-generation pytorch score-based-generative-modeling image2image text2image stable-diffusion stable-diffusion-diffusers hacktoberfest

huggingface / transformers
jax,🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Organization: huggingface
Home Page: https://huggingface.co/transformers
nlp natural-language-processing pytorch language-model tensorflow bert language-models pytorch-transformers nlp-library transformer

iree-org / iree
jax,A retargetable MLIR-based machine learning compiler and runtime toolkit.
Organization: iree-org
Home Page: http://iree.dev/
mlir vulkan tensorflow spirv cuda jax pytorch compiler machine-learning runtime

jermwatt / machine_learning_refined
jax,Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).
User: jermwatt
machine-learning deep-learning artificial-intelligence data-science jupyter-notebook numpy autograd machine-learning-algorithms lecture-notes slides

keras-team / keras
jax,Deep Learning for humans
Organization: keras-team
Home Page: http://keras.io/
deep-learning tensorflow neural-networks machine-learning data-science python jax pytorch

kubeflow / katib
jax,Automated Machine Learning on Kubernetes
Organization: kubeflow
Home Page: https://www.kubeflow.org/docs/components/katib
ai automl huggingface hyperparameter-tuning jax kubeflow kubernetes llm machine-learning mlops neural-architecture-search pytorch scikit-learn tensorflow

kubeflow / training-operator
jax,Distributed ML Training and Fine-Tuning on Kubernetes
Organization: kubeflow
Home Page: https://www.kubeflow.org/docs/components/training
ai distributed fine-tuning gpu huggingface jax kubeflow kubernetes llm machine-learning

lululxvi / deepxde
jax,A library for scientific machine learning and physics-informed learning
User: lululxvi
Home Page: https://deepxde.readthedocs.io
neural-network deep-learning scientific-machine-learning pinn multi-fidelity-data operator pytorch physics-informed-learning jax deeponet

ml-tooling / best-of-ml-python
jax,🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
Organization: ml-tooling
Home Page: https://ml-python.best-of.org
python machine-learning data-science nlp data-visualization tensorflow pytorch scikit-learn keras data-visualizations

n2cholas / awesome-jax
jax,JAX - A curated list of resources https://github.com/google/jax
User: n2cholas
awesome awesome-list jax machine-learning neural-network deep-learning xla autograd numpy

nvidia / transformerengine
jax,A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Organization: nvidia
Home Page: https://docs.nvidia.com/deeplearning/transformer-engine/user-guide/index.html
cuda deep-learning gpu machine-learning python pytorch fp8 jax

parrt / tensor-sensor
jax,The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
User: parrt
Home Page: https://github.com/parrt/tensor-sensor
numpy pytorch matrix vector debugging tracing deep-learning python tensorflow jax

patrick-kidger / diffrax
jax,Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
User: patrick-kidger
differential-equations jax deep-learning neural-differential-equations machine-learning dynamical-systems neural-networks equinox

patrick-kidger / equinox
jax,Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
User: patrick-kidger
deep-learning jax neural-networks equinox

patrick-kidger / jaxtyping
jax,Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
User: patrick-kidger
jax python-typing typing

pennylaneai / pennylane
jax,PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
Organization: pennylaneai
Home Page: https://pennylane.ai
quantum machine-learning deep-learning neural-network optimization quantum-computing quantum-machine-learning automatic-differentiation tensorflow pytorch

phlippe / uvadlc_notebooks
jax,Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
User: phlippe
Home Page: https://uvadlc-notebooks.readthedocs.io/en/latest/
deep-learning tutorials uvadlc pytorch pytorch-lightning tutorial flax jax optax

probml / pyprobml
jax,Python code for "Probabilistic Machine learning" book by Kevin Murphy
Organization: probml
jupyter-notebooks machine-learning colab pml probabilistic-programming jax tensorflow pytorch pymc3 numpyro

pyro-ppl / numpyro
jax,Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Organization: pyro-ppl
Home Page: https://num.pyro.ai
pyro jax hmc inference-algorithms numpy bayesian-inference probabilistic-programming mcmc

sanchit-gandhi / whisper-jax
jax,JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
User: sanchit-gandhi
deep-learning jax speech-recognition speech-to-text whisper

tensorflow / datasets
jax,TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Organization: tensorflow
Home Page: https://www.tensorflow.org/datasets
tensorflow machine-learning data datasets numpy jax dataset

tensorly / tensorly
jax,TensorLy: Tensor Learning in Python.
Organization: tensorly
Home Page: http://tensorly.org
machine-learning tensor decomposition tensor-algebra tensorly tensor-learning python tensor-methods pytorch mxnet

unifyai / ivy
jax,The Unified AI Framework
Organization: unifyai
Home Page: https://unify.ai
abstraction autograd deep-learning gpu ivy jax machine-learning mxnet neural-network numpy python pytorch template tensorflow

wandb / wandb
jax,🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Organization: wandb
Home Page: https://wandb.ai
machine-learning experiment-track deep-learning keras tensorflow pytorch hyperparameter-search reinforcement-learning mlops data-science

yang-song / score_sde
jax,Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
User: yang-song
Home Page: https://arxiv.org/abs/2011.13456
score-matching stochastic-differential-equations generative-models score-based-generative-modeling controllable-generation inverse-problems jax flax iclr-2021 diffusion-models

young-geng / easylm
jax,Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
User: young-geng
deep-learning flax jax language-model natural-language-processing transformer large-language-models chatbot llama