jax-ml/jax-ai-stack issues and pull requests

#163 - Bump orbax-checkpoint from 0.11.0 to 0.11.1

Pull Request - State: closed - Opened by dependabot[bot] 5 days ago
Labels: dependencies, python

#162 - Bump actions/setup-python from 5.3.0 to 5.4.0

Pull Request - State: closed - Opened by dependabot[bot] 5 days ago
Labels: dependencies, github_actions

#161 - Failed build: Nightly Tests

Issue - State: closed - Opened by github-actions[bot] 6 days ago - 2 comments
Labels: build failed

#160 - Build: set version dynamically in pyproject.toml

Pull Request - State: closed - Opened by jakevdp 7 days ago

#159 - Fix landing page links to docs.jaxstack.ai

Pull Request - State: closed - Opened by skye 10 days ago

#158 - Upgrade JAX AI Stack for miniGPT doc

Pull Request - State: open - Opened by 8bitmp3 10 days ago - 1 comment

#157 - Update tutorial: Visualize JAX model

Pull Request - State: open - Opened by pavithraes 12 days ago - 2 comments

#156 - Bump pypa/gh-action-pypi-publish from 1.12.2 to 1.12.4

Pull Request - State: closed - Opened by dependabot[bot] 12 days ago
Labels: dependencies, github_actions

#155 - Simplify landing page to static HTML + CSS

Pull Request - State: closed - Opened by skye 14 days ago - 1 comment

#154 - Discussion : What is the functionality of `pygrain.ShardOption`, like `pygrain.ShardByJaxProcess`?

Issue - State: open - Opened by carlesoctav 17 days ago

#153 - Bump jax from 0.4.38 to 0.5.0

Pull Request - State: closed - Opened by dependabot[bot] 19 days ago
Labels: dependencies, python

#152 - Small tweaks

Pull Request - State: closed - Opened by rcrowe-google 23 days ago - 1 comment

#151 - Add chex to dependencies

Pull Request - State: closed - Opened by jakevdp 24 days ago - 4 comments

#150 - Add chex

Issue - State: closed - Opened by jakevdp 24 days ago

#149 - Colab XLA Error in JAX_for_LLM_pretraining.ipynb

Issue - State: open - Opened by Matt-Hurd 25 days ago - 2 comments

#148 - Bump ml-dtypes from 0.4.0 to 0.5.1

Pull Request - State: closed - Opened by dependabot[bot] 26 days ago - 2 comments
Labels: dependencies, python

#147 - Bump actions/upload-artifact from 4.5.0 to 4.6.0

Pull Request - State: closed - Opened by dependabot[bot] 26 days ago
Labels: dependencies, github_actions

#146 - Fix typo in changelog

Pull Request - State: closed - Opened by jakevdp 28 days ago

#145 - Incorrect version of `grain` in releases

Issue - State: closed - Opened by Tomas542 28 days ago - 1 comment

#144 - CHANGELOG: use unicode arrow to indicate updated versions

Pull Request - State: closed - Opened by jakevdp 29 days ago

#143 - Bump version to 2025-1-9

Pull Request - State: closed - Opened by jakevdp 29 days ago

#142 - Update JAX AI Stack Diffusion tutorial

Pull Request - State: open - Opened by 8bitmp3 about 1 month ago - 1 comment

#141 - grain->0.2.3 orbax-checkpoint->0.11.0

Pull Request - State: closed - Opened by jakevdp about 1 month ago

#140 - Bump grain from 0.2.2 to 0.2.3

Pull Request - State: closed - Opened by dependabot[bot] about 1 month ago - 1 comment
Labels: dependencies, python

#139 - Bump actions/upload-artifact from 4.4.3 to 4.5.0

Pull Request - State: closed - Opened by dependabot[bot] about 2 months ago
Labels: dependencies, github_actions

#138 - Bump jax from 0.4.37 to 0.4.38

Pull Request - State: closed - Opened by dependabot[bot] about 2 months ago
Labels: dependencies, python

#137 - Bump orbax-checkpoint from 0.10.2 to 0.10.3

Pull Request - State: closed - Opened by dependabot[bot] about 2 months ago - 2 comments
Labels: dependencies, python

#136 - Sidebar restructure and minor clean-up

Pull Request - State: closed - Opened by pavithraes about 2 months ago

#135 - Update JAX AI for PyTorch users tutorial

Pull Request - State: open - Opened by 8bitmp3 about 2 months ago

#134 - Failed build: Nightly Tests

Issue - State: closed - Opened by github-actions[bot] about 2 months ago - 24 comments
Labels: build failed

#133 - Add Llama 3 tutorial (porting HF model to JAX)

Pull Request - State: open - Opened by windmaple about 2 months ago - 4 comments

#132 - minor refresh to dev docs needed

Issue - State: closed - Opened by windmaple about 2 months ago

#131 - Bump pypa/gh-action-pypi-publish from 1.12.2 to 1.12.3

Pull Request - State: closed - Opened by dependabot[bot] about 2 months ago - 1 comment
Labels: dependencies, github_actions

#130 - bump version to 2024.12.10

Pull Request - State: closed - Opened by jakevdp about 2 months ago

#129 - bump JAX to v0.4.37

Pull Request - State: closed - Opened by jakevdp about 2 months ago

#128 - Bump orbax-checkpoint from 0.10.1 to 0.10.2

Pull Request - State: closed - Opened by dependabot[bot] 2 months ago
Labels: dependencies, python

#127 - Bump jax from 0.4.35 to 0.4.36

Pull Request - State: closed - Opened by dependabot[bot] 2 months ago
Labels: dependencies, python

#126 - Update to absolute links to doc sites

Pull Request - State: closed - Opened by MichaelHudgins 2 months ago

#125 - Add install guide

Pull Request - State: closed - Opened by jakevdp 2 months ago

#124 - Add open in colab links to docs

Pull Request - State: closed - Opened by jakevdp 2 months ago

#123 - Fix heading in JAX_for_LLM_pretraining tutorial

Pull Request - State: closed - Opened by windmaple 2 months ago

#122 - Should all 'open in colab' buttons be removed in docs/*

Issue - State: closed - Opened by windmaple 2 months ago - 2 comments

#121 - Adds styling, base content, new template

Pull Request - State: closed - Opened by jakevdp 2 months ago

#120 - Question- Why Chex isn't part of jax-ai-stack?

Issue - State: open - Opened by eadadi 2 months ago

#119 - Add Llama 3 tutorial (porting HF model to JAX)

Pull Request - State: closed - Opened by windmaple 2 months ago - 5 comments

#118 - Adding tensorboard tracking/visualization and nnx display example

Pull Request - State: closed - Opened by dillonroach 2 months ago - 3 comments

#117 - Added Image caption tutorial

Pull Request - State: closed - Opened by vfdev-5 2 months ago - 2 comments

#116 - Update ViT tutorial

Pull Request - State: closed - Opened by vfdev-5 2 months ago

#115 - Fixed linter

Pull Request - State: closed - Opened by vfdev-5 2 months ago

#114 - ENH - Add logo and favicon to docs

Pull Request - State: closed - Opened by trallard 2 months ago - 5 comments

#113 - Fixed URL typo

Pull Request - State: closed - Opened by vfdev-5 2 months ago

#112 - Bump orbax-checkpoint from 0.9.1 to 0.10.1

Pull Request - State: closed - Opened by dependabot[bot] 2 months ago - 1 comment
Labels: dependencies, python

#111 - Bump flax from 0.10.1 to 0.10.2

Pull Request - State: closed - Opened by dependabot[bot] 2 months ago - 1 comment
Labels: dependencies, python

#110 - Adding tutorial for data loaders in multi device setups

Pull Request - State: open - Opened by selamw1 3 months ago - 9 comments

#109 - Adding tutorial for data loaders on gpu with jax

Pull Request - State: closed - Opened by selamw1 3 months ago - 9 comments

#108 - Adding tutorial for data loaders on cpu with jax

Pull Request - State: closed - Opened by selamw1 3 months ago - 7 comments

#107 - NLP: JAX Machine Translation

Pull Request - State: closed - Opened by dillonroach 3 months ago - 2 comments

#106 - Lint JAX AI Stack Getting Started

Pull Request - State: closed - Opened by 8bitmp3 3 months ago

#105 - JAX time series classification

Pull Request - State: closed - Opened by mtsokol 3 months ago - 4 comments

#104 - CI: add testing on Python 3.13

Pull Request - State: closed - Opened by jakevdp 3 months ago - 2 comments

#103 - JAX transformer text classification

Pull Request - State: closed - Opened by mtsokol 3 months ago - 3 comments

#102 - Added ViT tutorial

Pull Request - State: closed - Opened by vfdev-5 3 months ago - 2 comments

#101 - Change documentation theme

Pull Request - State: closed - Opened by pavithraes 3 months ago

#100 - Update package versions

Pull Request - State: closed - Opened by jakevdp 3 months ago

#99 - Bump optax from 0.2.3 to 0.2.4

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago - 1 comment
Labels: dependencies, python

#98 - Bump orbax-export from 0.0.5 to 0.0.6

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago - 1 comment
Labels: dependencies, python

#97 - Bump orbax-checkpoint from 0.8.0 to 0.9.1

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago - 1 comment
Labels: dependencies, python

#96 - Update JAX AI Stack VAE tutorial

Pull Request - State: closed - Opened by 8bitmp3 3 months ago - 1 comment

#95 - Added NLP: Basic text classification with JAX & Flax

Pull Request - State: closed - Opened by vfdev-5 3 months ago - 2 comments

#94 - Added badge link to readthedocs site: https://jax-ai-stack.readthedocs.io/en/latest/index.html

Pull Request - State: closed - Opened by vfdev-5 3 months ago

#93 - Added JAX domain example, vision image segmentation

Pull Request - State: closed - Opened by vfdev-5 3 months ago - 9 comments

#92 - Bump pypa/gh-action-pypi-publish from 1.11.0 to 1.12.2

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, github_actions

#91 - Bump orbax-checkpoint from 0.8.0 to 0.9.0

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago - 2 comments
Labels: dependencies, python

#90 - Failed build: Nightly Tests

Issue - State: closed - Opened by github-actions[bot] 3 months ago - 24 comments
Labels: build failed

#89 - Add Profiling Section to LLM Pretraining Tutorial

Pull Request - State: closed - Opened by Matt-Hurd 3 months ago

#88 - Minor fix in JAX_porting_PyTorch_model tutorial

Pull Request - State: closed - Opened by vfdev-5 3 months ago

#87 - Bump pypa/gh-action-pypi-publish from 1.10.3 to 1.11.0

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, github_actions

#86 - Bump tensorflow-datasets from 4.9.6 to 4.9.7

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, python

#85 - Bump orbax-checkpoint from 0.7.0 to 0.8.0

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, python

#84 - Bump flax from 0.10.0 to 0.10.1

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, python

#83 - Add the simple diffusion model tutorial

Pull Request - State: closed - Opened by lucianommartins 3 months ago - 1 comment

#82 - Bump version to 2024.11.1

Pull Request - State: closed - Opened by jakevdp 3 months ago

#81 - Add LLM pretraining tutorial

Pull Request - State: closed - Opened by windmaple 3 months ago - 4 comments
Labels: documentation

#80 - Bump actions/checkout from 4.2.1 to 4.2.2

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, github_actions

#79 - Bump actions/setup-python from 5.2.0 to 5.3.0

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago - 2 comments
Labels: dependencies, github_actions

#78 - Bump jax from 0.4.34 to 0.4.35

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, python

#77 - Bump tensorflow from 2.17.0 to 2.18.0

Pull Request - State: closed - Opened by dependabot[bot] 3 months ago
Labels: dependencies, python

#76 - Bump grain from 0.2.1 to 0.2.2

Pull Request - State: closed - Opened by dependabot[bot] 4 months ago
Labels: dependencies, python

#75 - Fix typos in VAE tutorial

Pull Request - State: closed - Opened by jakevdp 4 months ago
Labels: documentation

#74 - Add linkify extension to automatically create links

Pull Request - State: closed - Opened by melissawm 4 months ago

#73 - DOC: Update link to Getting Started guide in index page

Pull Request - State: closed - Opened by melissawm 4 months ago - 1 comment

#72 - Impossible to install [grain] on Grace Hopper with Python 3.12 due to non-existing grain for aarch64 for python 3.12

Issue - State: closed - Opened by surak 4 months ago - 1 comment

GitHub / jax-ml/jax-ai-stack issues and pull requests