Movatterモバイル変換

Skip to content

#

jax

Here are 1,314 public repositories matching this topic...

Language:All

Filter by language

All1,314 Python948 Jupyter Notebook283 C++10 HTML8 Shell7 Dockerfile4 TeX4 C2 Java2 AppleScript1

Sort:Most stars

Sort options

Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

keras-team /keras

Deep Learning for humans

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

UpdatedJul 17, 2025
Python

jax

jax-ml /jax

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

UpdatedJul 18, 2025
Python

huggingface /diffusers

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

flux deep-learning pytorch image-generation flax hacktoberfest diffusion text2image image2image jax text2video score-based-generative-modeling stable-diffusion stable-diffusion-diffusers latent-diffusion-models video2video image2video

UpdatedJul 18, 2025
Python

d2l-ai /d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 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

UpdatedAug 18, 2024
Python

lukasmasuch /best-of-ml-python

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

python nlp data-science machine-learning deep-learning tensorflow scikit-learn keras ml data-visualization pytorch transformer data-analysis gpt automl jax data-visualizations gpt-3 chatgpt

UpdatedJul 17, 2025

ivy-llc /ivy

Convert Machine Learning Code Between Frameworks

python tensorflow numpy pytorch jax

UpdatedJul 18, 2025
Python

wandb

wandb /wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

data-science machine-learning reinforcement-learning ai deep-learning tensorflow keras collaboration pytorch hyperparameter-optimization reproducibility hyperparameter-tuning data-versioning hyperparameter-search model-versioning jax mlops experiment-track ml-platform

UpdatedJul 18, 2025
Python

einops

arogozhnikov /einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

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

UpdatedJul 1, 2025
Python

google /trax

Trax — Deep Learning with Clear Code and Speed

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

UpdatedApr 10, 2025
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

UpdatedNov 26, 2024
Jupyter Notebook

google /flax

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

UpdatedJul 18, 2025
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

UpdatedJan 21, 2022

sanchit-gandhi /whisper-jax

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

deep-learning speech-recognition speech-to-text whisper jax

UpdatedApr 3, 2024
Jupyter Notebook

tensorflow /datasets

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

data machine-learning tensorflow numpy dataset datasets jax

UpdatedJul 17, 2025
Python

google-research /scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

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

UpdatedJul 9, 2025
Python

lululxvi /deepxde

A library for scientific machine learning and physics-informed learning

deep-learning neural-network tensorflow pytorch operator pde paddle pinn jax scientific-machine-learning multi-fidelity-data physics-informed-learning deeponet

UpdatedJun 21, 2025
Python

iree-org /iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

machine-learning compiler runtime tensorflow vulkan cuda pytorch spirv onnx jax mlir

UpdatedJul 18, 2025
C++

alpa-projects /alpa

Training and serving large-scale neural networks with auto parallelization.

machine-learning deep-learning compiler distributed-computing high-performance-computing distributed-training jax alpa auto-parallelization llm

UpdatedDec 9, 2023
Python

google-deepmind /dm-haiku

JAX-based neural network library

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

UpdatedJul 13, 2025
Python

phlippe /uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

tutorial deep-learning tutorials pytorch flax jax pytorch-lightning optax uvadlc

UpdatedMar 16, 2025
Jupyter Notebook

Improve this page

Add a description, image, and links to thejax 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 thejax topic, visit your repo's landing page and select "manage topics."

[8]ページ先頭

©2009-2025 Movatter.jp