github-starred/awesome-python
2
0
Fork 0
mirror of https://github.com/vinta/awesome-python.git synced 2026-03-22 14:12:18 -05:00
Code Issues 497 Packages Projects Releases Wiki Activity

[PR #2612] add jax #1929

New Issue
Open
opened 2025-11-06 13:26:02 -06:00 by GiteaMirror · 0 comments
GiteaMirror commented 2025-11-06 13:26:02 -06:00
Owner
Copy Link

📋 Pull Request Information

Original PR: https://github.com/vinta/awesome-python/pull/2612
Author: @wanikhawar
Created: 9/5/2024
Status: 🔄 Open

Base: master ← Head: patch-1

📝 Commits (1)

📊 Changes

1 file changed (+1 additions, -0 deletions)

View changed files

📝 README.md (+1 -0)

📄 Description

What is this Python project?

JAX is a high-performance library designed for array-oriented numerical computation. It offers automatic differentiation and Just-In-Time (JIT) compilation, making it highly suitable for machine learning research and other computationally intensive tasks.

Features

  • Unified interface: JAX offers a NumPy-like interface for computations that can seamlessly run on CPUs, GPUs, or TPUs, and scale across local or distributed environments.

  • JIT compilation: JAX includes built-in Just-In-Time (JIT) compilation through OpenXLA, an open-source machine learning compiler framework.

  • Automatic differentiation: JAX efficiently computes gradients through its automatic differentiation capabilities, making it ideal for optimization and machine learning tasks.

  • Automatic vectorization: JAX supports automatic vectorization, enabling efficient computation over batches of inputs by applying functions across array elements in parallel.

What's the difference between this Python project and similar ones?

  • NumPy Compatibility: JAX provides a NumPy-like interface but extends its functionality with automatic differentiation and GPU/TPU support, capabilities not present in standard NumPy.

  • Comparison with TensorFlow/PyTorch: While TensorFlow and PyTorch are popular frameworks, JAX offers more fine-grained control over the computational graph and is based on functional programming, which enhances flexibility for research and experimentation.

--

Anyone who agrees with this pull request could submit an Approve review to it.

🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.

## 📋 Pull Request Information **Original PR:** https://github.com/vinta/awesome-python/pull/2612 **Author:** [@wanikhawar](https://github.com/wanikhawar) **Created:** 9/5/2024 **Status:** 🔄 Open **Base:** `master` ← **Head:** `patch-1` --- ### 📝 Commits (1) - [`247f5ce`](https://github.com/vinta/awesome-python/commit/247f5ceb507892aafaea32c0cb971c4a9abbbb7d) add jax ### 📊 Changes **1 file changed** (+1 additions, -0 deletions) <details> <summary>View changed files</summary> 📝 `README.md` (+1 -0) </details> ### 📄 Description ## What is this Python project? JAX is a high-performance library designed for array-oriented numerical computation. It offers automatic differentiation and Just-In-Time (JIT) compilation, making it highly suitable for machine learning research and other computationally intensive tasks. ### Features - Unified interface: JAX offers a NumPy-like interface for computations that can seamlessly run on CPUs, GPUs, or TPUs, and scale across local or distributed environments. - JIT compilation: JAX includes built-in Just-In-Time (JIT) compilation through [OpenXLA](https://github.com/openxla), an open-source machine learning compiler framework. - Automatic differentiation: JAX efficiently computes gradients through its automatic differentiation capabilities, making it ideal for optimization and machine learning tasks. - Automatic vectorization: JAX supports automatic vectorization, enabling efficient computation over batches of inputs by applying functions across array elements in parallel. ## What's the difference between this Python project and similar ones? - NumPy Compatibility: JAX provides a NumPy-like interface but extends its functionality with automatic differentiation and GPU/TPU support, capabilities not present in standard NumPy. - Comparison with TensorFlow/PyTorch: While TensorFlow and PyTorch are popular frameworks, JAX offers more fine-grained control over the computational graph and is based on functional programming, which enhances flexibility for research and experimentation. -- Anyone who agrees with this pull request could submit an *Approve* review to it. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
GiteaMirror added the pull-request label 2025-11-06 13:26:02 -06:00
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
pull-request
Mirrored from GitHub Pull Request
No Label pull-request
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
GiteaMirror ninjasurge
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: github-starred/awesome-python#1929
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?