Skip to content

Navigation Menu

Appearance settings

GitHub Copilot
Write better code with AI

GitHub Spark New
Build and deploy intelligent apps

GitHub Models New
Manage and compare prompts

GitHub Advanced Security
Find and fix vulnerabilities

Actions
Automate any workflow
Codespaces
Instant dev environments

Issues
Plan and track work

Code Review
Manage code changes

Discussions
Collaborate outside of code

Code Search
Find more, search less
Explore

Why GitHub

Documentation

GitHub Skills

Blog
Integrations

GitHub Marketplace

MCP Registry
View all features
By company size

Enterprises

Small and medium teams

Startups

Nonprofits
By use case

App Modernization

DevSecOps

DevOps

CI/CD

View all use cases
By industry

Healthcare

Financial services

Manufacturing

Government

View all industries
View all solutions
Topics

AI

DevOps

Security

Software Development

View all
Explore

Learning Pathways

Events & Webinars

Ebooks & Whitepapers

Customer Stories

Partners

Executive Insights
GitHub Sponsors
Fund open source developers
The ReadME Project
GitHub community articles
Repositories

Topics

Trending

Collections
Enterprise platform
AI-powered developer platform
Available add-ons

GitHub Advanced Security
Enterprise-grade security features

Copilot for business
Enterprise-grade AI features

Premium Support
Enterprise-grade 24/7 support
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.

Appearance settings

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 Public

Notifications You must be signed in to change notification settings
Fork 3.2k
Star 33.6k

Code
Issues 1.6k
Pull requests 565
Discussions
Actions
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Discussions
Actions
Security
Insights

Pull requests: jax-ml/jax

Labels 54 Milestones 1

Labels 54 Milestones 1

New pull request New

Clear current search query, filters, and sorts

5 Open 55 Closed

5 Open 55 Closed

Author

Filter by author

Loading

Uh oh!

There was an error while loading. Please reload this page.

Label

Filter by label

Loading

Uh oh!

There was an error while loading. Please reload this page.

Use alt + click/return to exclude labels

or ⇧ + click/return for logical OR

Projects

Filter by project

Loading

Uh oh!

There was an error while loading. Please reload this page.

Milestones

Filter by milestone

Loading

Uh oh!

There was an error while loading. Please reload this page.

Reviews

Filter by reviews

No reviews Review required Approved review Changes requested

Assignee

Filter by who’s assigned

Assigned to nobody

Loading

Uh oh!

There was an error while loading. Please reload this page.

Sort

Sort by

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

Most reactions

Pull requests list

[WIP] Implemented an extract_image_patches op using lax ops. cla: yes

#1154 by gehring was closed Oct 12, 2020

Loading…

[Waiting Reviews]: Fixes the parameters descriptions cla: yes

#1434 by joaogui1 was merged Oct 30, 2019

Loading…

WIP: Add a pylintrc to make it easire to use linter cla: yes

#1442 by joaogui1 was merged Oct 5, 2019

Loading…

8

Implementation of np.ldexp and np.frexp cla: yes

#1529 by WindQAQ was merged Apr 1, 2020

Loading…

Adding broadcast_argnums to pmap for allowing similar behaviour t… cla: yes

#1786 by botev was merged Feb 14, 2020

Loading…

Fixed the documentation of api.linearize