Skip to content

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.

Sign in

Sign up

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 2.8k
Star 30.1k

Code
Issues 1.4k
Pull requests 379
Discussions
Actions
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Discussions
Actions
Security
Insights

CI

[JAX] Make a one hot mode of take along axis. #62876

Sign in to view logs

Sign in to view logs

[JAX] Make a one hot mode of take along axis.

[JAX] Make a one hot mode of take along axis. #62876

Summary
Jobs
Run details
- Usage
- Workflow file

Usage Workflow file

Documentation - test code snippets (3.10)

succeeded Oct 1, 2024 in 1m 30s

Loading

Re-running jobs...

You can’t perform that action at this time.