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

google-deepmind / optax Public

Notifications You must be signed in to change notification settings
Fork 199
Star 1.8k

Code
Issues 39
Pull requests 25
Discussions
Actions
Projects
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Discussions
Actions
Projects
Security
Insights

tests

Support tuple of axis in `softmax_cross_entropy_with_integer_labels` #5128

Sign in to view logs

Sign in to view logs

Support tuple of axis in `softmax_cross_entropy_with_integer_labels`

Support tuple of axis in `softmax_cross_entropy_with_integer_labels` #5128

Summary
Jobs
Run details
- Usage
- Workflow file

Usage Workflow file

Annotations

2 errors

The job was canceled because "_3_11_ubuntu-latest_newes" failed.

Run CI tests

The operation was canceled.

Python 3.9 on ubuntu-latest jax=0.4.27

Python 3.9 on ubuntu-latest jax=0.4.27

cancelled Jan 2, 2025 in 16m 59s

Loading

Re-running jobs...

You can’t perform that action at this time.