Movatterモバイル変換

Skip to content

#

jax

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

Language:All

Filter by language

All1,194 Python852 Jupyter Notebook267 C++8 HTML7 Shell7 TeX4 Dockerfile3 C2 Java2 AppleScript1

Sort:Most stars

Sort options

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

transformers

huggingface /transformers

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

python nlp machine-learning natural-language-processing deep-learning tensorflow pytorch transformer speech-recognition seq2seq flax pretrained-models language-models nlp-library language-model hacktoberfest bert jax pytorch-transformers model-hub

UpdatedMar 17, 2025
Python

keras-team /keras

Deep Learning for humans

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

UpdatedMar 17, 2025
Python

jax

jax-ml /jax

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

UpdatedMar 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

UpdatedMar 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

ml-tooling /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

UpdatedMar 13, 2025

ivy-llc /ivy

Convert Machine Learning Code Between Frameworks

python converter machine-learning translation deep-learning neural-network tensorflow numpy pytorch ivy transpilation jax

UpdatedFeb 21, 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

UpdatedMar 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

UpdatedFeb 9, 2025
Python

google /trax

Trax — Deep Learning with Clear Code and Speed

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

UpdatedFeb 7, 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.

UpdatedMar 18, 2025
Jupyter Notebook

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

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

tensorflow /datasets

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

data machine-learning tensorflow numpy dataset datasets jax

UpdatedMar 14, 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

UpdatedMar 12, 2025
Python

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

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

UpdatedMar 18, 2025
C++

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

UpdatedMar 14, 2025
Python

google-deepmind /dm-haiku

JAX-based neural network library

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

UpdatedMar 13, 2025
Python

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