Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

jax-ml / jax-ai-stack Public

Notifications You must be signed in to change notification settings
Fork 22
Star 114

Code
Issues 3
Pull requests 6
Actions
Projects
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Security
Insights

Pull requests: jax-ml/jax-ai-stack

Labels 13 Milestones 0

Labels 13 Milestones 0

New pull request New

6 Open 140 Closed

6 Open 140 Closed

Author

Filter by author

Loading

Label

Filter by label

Loading

Use alt + click/return to exclude labels

or ⇧ + click/return for logical OR

Projects

Filter by project

Loading

Milestones

Filter by milestone

Loading

Reviews

Filter by reviews

No reviews Review required Approved review Changes requested

Assignee

Filter by who’s assigned

Assigned to nobody

Loading

Sort

Sort by

Newest Oldest Most commented Least commented Recently updated Least recently updated Best match

Most reactions

Pull requests list

Upgrade JAX AI Stack for miniGPT doc

#158 opened Jan 28, 2025 by 8bitmp3 • Draft

Update tutorial: Visualize JAX model

#157 opened Jan 27, 2025 by pavithraes

Loading…

1

Update JAX AI Stack Diffusion tutorial

#142 opened Jan 2, 2025 by 8bitmp3 • Draft

1

Update JAX AI for PyTorch users tutorial

#135 opened Dec 20, 2024 by 8bitmp3 • Draft

1

Add Llama 3 tutorial (porting HF model to JAX)

#133 opened Dec 17, 2024 by windmaple

Loading…

4

Adding tutorial for data loaders in multi device setups

#110 opened Nov 21, 2024 by selamw1

Loading…

ProTip! Exclude everything labeled bug with -label:bug.

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.