René's URL Explorer Experiment


Title: Add JAX kmeans implementation by gileshd · Pull Request #371 · probml/dynamax · GitHub

Open Graph Title: Add JAX kmeans implementation by gileshd · Pull Request #371 · probml/dynamax

X Title: Add JAX kmeans implementation by gileshd · Pull Request #371 · probml/dynamax

Description: A Python package for probabilistic state space modeling with JAX - Add JAX kmeans implementation by gileshd · Pull Request #371 · probml/dynamax

Open Graph Description: Summary This PR add an implementation of the k-means algorithm written in JAX. By replacing the current calls to the sklearn implementation with this new implementation, we can now vmap and jit cod...

X Description: Summary This PR add an implementation of the k-means algorithm written in JAX. By replacing the current calls to the sklearn implementation with this new implementation, we can now vmap and jit cod...

Opengraph URL: https://github.com/probml/dynamax/pull/371

X: @github

direct link

Domain: github.com

route-pattern/_view_fragments/voltron/pull_requests/show/:user_id/:repository/:id/pull_request_layout(.:format)
route-controllervoltron_pull_requests_fragments
route-actionpull_request_layout
fetch-noncev2:d0de7e39-af8d-6851-e8b3-7da75daafe06
current-catalog-service-hashae870bc5e265a340912cde392f23dad3671a0a881730ffdadd82f2f57d81641b
request-idE45A:CFFC3:21B58B7:30B0AF0:6A4BDAB0
html-safe-nonce1a20d1871224e06f5af1a183952826bddbcc5e84db705ce667731f0c0a3e4c56
visitor-payloadeyJyZWZlcnJlciI6IiIsInJlcXVlc3RfaWQiOiJFNDVBOkNGRkMzOjIxQjU4Qjc6MzBCMEFGMDo2QTRCREFCMCIsInZpc2l0b3JfaWQiOiI4ODY2MzczOTk2MjE3MzU5MDI0IiwicmVnaW9uX2VkZ2UiOiJpYWQiLCJyZWdpb25fcmVuZGVyIjoiaWFkIn0=
visitor-hmac93f9b0119b673d998cf19f92657f69e56c231067edeaef5d3cd621956f5cf9c6
hovercard-subject-tagpull_request:1980029460
github-keyboard-shortcutsrepository,pull-request-list,pull-request-conversation,pull-request-files-changed,copilot
google-site-verificationApib7-x98H0j5cPqHWwSMm6dNU4GmODRoqxLiDzdx9I
octolytics-urlhttps://collector.github.com/github/collect
analytics-location///voltron/pull_requests_fragments/pull_request_layout
fb:app_id1401488693436528
apple-itunes-appapp-id=1477376905, app-argument=https://github.com/_view_fragments/voltron/pull_requests/show/probml/dynamax/371/pull_request_layout
twitter:imagehttps://opengraph.githubassets.com/e942a1ca895855b49b53e86fbfb4cd167b6a94e9e9314e0beaa7445b63750101/probml/dynamax/pull/371
twitter:cardsummary_large_image
og:imagehttps://opengraph.githubassets.com/e942a1ca895855b49b53e86fbfb4cd167b6a94e9e9314e0beaa7445b63750101/probml/dynamax/pull/371
og:image:altSummary This PR add an implementation of the k-means algorithm written in JAX. By replacing the current calls to the sklearn implementation with this new implementation, we can now vmap and jit cod...
og:image:width1200
og:image:height600
og:site_nameGitHub
og:typeobject
og:author:usernamegileshd
hostnamegithub.com
expected-hostnamegithub.com
None9df996be9551ac02247d09a2f7f64ece66c35ca28885d346e98fdfda9cdaa37b
turbo-cache-controlno-cache
go-importgithub.com/probml/dynamax git https://github.com/probml/dynamax.git
octolytics-dimension-user_id6309387
octolytics-dimension-user_loginprobml
octolytics-dimension-repository_id480587737
octolytics-dimension-repository_nwoprobml/dynamax
octolytics-dimension-repository_publictrue
octolytics-dimension-repository_is_forkfalse
octolytics-dimension-repository_network_root_id480587737
octolytics-dimension-repository_network_root_nwoprobml/dynamax
turbo-body-classeslogged-out env-production page-responsive
disable-turbofalse
browser-stats-urlhttps://api.github.com/_private/browser/stats
browser-errors-urlhttps://api.github.com/_private/browser/errors
release7242837c61fe4c8b774b83b9ee05b1881ba589f8
ui-targetfull
theme-color#1e2327
color-schemelight dark

Links:

Skip to contenthttps://github.com/probml/dynamax/pull/371#start-of-content
https://github.com/
Sign in https://github.com/login?return_to=https%3A%2F%2Fgithub.com%2Fprobml%2Fdynamax%2Fpull%2F371
GitHub CopilotWrite better code with AIhttps://github.com/features/copilot
GitHub Copilot appDirect agents from issue to mergehttps://github.com/features/ai/github-app
MCP RegistryNewIntegrate external toolshttps://github.com/mcp
ActionsAutomate any workflowhttps://github.com/features/actions
CodespacesInstant dev environmentshttps://github.com/features/codespaces
IssuesPlan and track workhttps://github.com/features/issues
Code ReviewManage code changeshttps://github.com/features/code-review
GitHub Advanced SecurityFind and fix vulnerabilitieshttps://github.com/security/advanced-security
Code securitySecure your code as you buildhttps://github.com/security/advanced-security/code-security
Secret protectionStop leaks before they starthttps://github.com/security/advanced-security/secret-protection
Why GitHubhttps://github.com/why-github
Documentationhttps://docs.github.com
Bloghttps://github.blog
Changeloghttps://github.blog/changelog
Marketplacehttps://github.com/marketplace
View all featureshttps://github.com/features
Enterpriseshttps://github.com/enterprise
Small and medium teamshttps://github.com/team
Startupshttps://github.com/enterprise/startups
Nonprofitshttps://github.com/solutions/industry/nonprofits
App Modernizationhttps://github.com/solutions/use-case/app-modernization
DevSecOpshttps://github.com/solutions/use-case/devsecops
DevOpshttps://github.com/solutions/use-case/devops
CI/CDhttps://github.com/solutions/use-case/ci-cd
View all use caseshttps://github.com/solutions/use-case
Healthcarehttps://github.com/solutions/industry/healthcare
Financial serviceshttps://github.com/solutions/industry/financial-services
Manufacturinghttps://github.com/solutions/industry/manufacturing
Governmenthttps://github.com/solutions/industry/government
View all industrieshttps://github.com/solutions/industry
View all solutionshttps://github.com/solutions
AIhttps://github.com/resources/articles?topic=ai
Software Developmenthttps://github.com/resources/articles?topic=software-development
DevOpshttps://github.com/resources/articles?topic=devops
Securityhttps://github.com/resources/articles?topic=security
View all topicshttps://github.com/resources/articles
Customer storieshttps://github.com/customer-stories
Events & webinarshttps://github.com/resources/events
Ebooks & reportshttps://github.com/resources/whitepapers
Business insightshttps://github.com/solutions/executive-insights
GitHub Skillshttps://skills.github.com
Documentationhttps://docs.github.com
Customer supporthttps://support.github.com
Community forumhttps://github.com/orgs/community/discussions
Trust centerhttps://github.com/trust-center
Partnershttps://github.com/partners
View all resourceshttps://github.com/resources
GitHub SponsorsFund open source developershttps://github.com/sponsors
Security Labhttps://securitylab.github.com
Maintainer Communityhttps://maintainers.github.com
Acceleratorhttps://github.com/accelerator
GitHub Starshttps://stars.github.com
Archive Programhttps://archiveprogram.github.com
Topicshttps://github.com/topics
Trendinghttps://github.com/trending
Collectionshttps://github.com/collections
Enterprise platformAI-powered developer platformhttps://github.com/enterprise
GitHub Advanced SecurityEnterprise-grade security featureshttps://github.com/security/advanced-security
Copilot for BusinessEnterprise-grade AI featureshttps://github.com/features/copilot/copilot-business
Premium SupportEnterprise-grade 24/7 supporthttps://github.com/premium-support
Pricinghttps://github.com/pricing
Search syntax tipshttps://docs.github.com/search-github/github-code-search/understanding-github-code-search-syntax
documentationhttps://docs.github.com/search-github/github-code-search/understanding-github-code-search-syntax
Sign in https://github.com/login?return_to=https%3A%2F%2Fgithub.com%2Fprobml%2Fdynamax%2Fpull%2F371
Sign up https://github.com/signup?ref_cta=Sign+up&ref_loc=header+logged+out&ref_page=%2F%3Cuser-name%3E%2F%3Crepo-name%3E%2Fvoltron%2Fpull_requests_fragments%2Fpull_request_layout&source=header-repo&source_repo=probml%2Fdynamax
Reloadhttps://github.com/probml/dynamax/pull/371
Reloadhttps://github.com/probml/dynamax/pull/371
Reloadhttps://github.com/probml/dynamax/pull/371
Please reload this pagehttps://github.com/probml/dynamax/pull/371
probml https://github.com/probml
dynamaxhttps://github.com/probml/dynamax
Notifications https://github.com/login?return_to=%2Fprobml%2Fdynamax
Fork 113 https://github.com/login?return_to=%2Fprobml%2Fdynamax
Star 978 https://github.com/login?return_to=%2Fprobml%2Fdynamax
Code https://github.com/probml/dynamax
Issues 56 https://github.com/probml/dynamax/issues
Pull requests 22 https://github.com/probml/dynamax/pulls
Actions https://github.com/probml/dynamax/actions
Projects https://github.com/probml/dynamax/projects
Security and quality 0 https://github.com/probml/dynamax/security
Insights https://github.com/probml/dynamax/pulse
Code https://github.com/probml/dynamax
Issues https://github.com/probml/dynamax/issues
Pull requests https://github.com/probml/dynamax/pulls
Actions https://github.com/probml/dynamax/actions
Projects https://github.com/probml/dynamax/projects
Security and quality https://github.com/probml/dynamax/security
Insights https://github.com/probml/dynamax/pulse
gileshdhttps://github.com/gileshd
mainhttps://github.com/probml/dynamax/tree/main
ghd/kmeans-refactorhttps://github.com/probml/dynamax/tree/ghd/kmeans-refactor
Conversationhttps://github.com/probml/dynamax/pull/371
Commits3 (3)https://github.com/probml/dynamax/pull/371/commits
Checkshttps://github.com/probml/dynamax/pull/371/checks
Files changedhttps://github.com/probml/dynamax/pull/371/files
Add JAX kmeans implementation https://github.com/probml/dynamax/pull/371#top
gileshdhttps://github.com/gileshd
mainhttps://github.com/probml/dynamax/tree/main
ghd/kmeans-refactorhttps://github.com/probml/dynamax/tree/ghd/kmeans-refactor
https://github.com/gileshd
gileshdhttps://github.com/gileshd
Jul 21, 2024https://github.com/probml/dynamax/pull/371#issue-2421507953
Please reload this pagehttps://github.com/probml/dynamax/pull/371
#315https://github.com/probml/dynamax/issues/315
Please reload this pagehttps://github.com/probml/dynamax/pull/371
gileshdhttps://github.com/gileshd
July 19, 2024 22:12https://github.com/probml/dynamax/pull/371#commits-pushed-420418e
https://github.com/gileshd
Add utility function for sklearn kmeanshttps://github.com/probml/dynamax/pull/371/commits/420418edf0d266dfeda9f8ff514a94638421000d
420418ehttps://github.com/probml/dynamax/pull/371/commits/420418edf0d266dfeda9f8ff514a94638421000d
https://github.com/gileshd
Update SSMs to use kmeans utility functionhttps://github.com/probml/dynamax/pull/371/commits/828f1d144e3631cbc7d9e790e11e9c83863e4242
828f1d1https://github.com/probml/dynamax/pull/371/commits/828f1d144e3631cbc7d9e790e11e9c83863e4242
https://github.com/gileshd
Add jax implementation of kmeanshttps://github.com/probml/dynamax/pull/371/commits/0f8646b45e67e89212f974ba5d2c9901fb887610
0f8646bhttps://github.com/probml/dynamax/pull/371/commits/0f8646b45e67e89212f974ba5d2c9901fb887610
https://github.com/gileshd
gileshdhttps://github.com/gileshd
slindermanhttps://github.com/slinderman
July 21, 2024 16:18https://github.com/probml/dynamax/pull/371#event-13593080906
https://github.com/gileshd
gileshdhttps://github.com/gileshd
Jul 21, 2024 https://github.com/probml/dynamax/pull/371#pullrequestreview-2190364977
View reviewed changes https://github.com/probml/dynamax/pull/371/files/0f8646b45e67e89212f974ba5d2c9901fb887610
dynamax/hidden_markov_model/models/logreg_hmm.pyhttps://github.com/probml/dynamax/pull/371/files/0f8646b45e67e89212f974ba5d2c9901fb887610#diff-486ed9ece7d65ebc15d353351dc8281e7fc24568b2b6c6f03d88f1d2f27a1578
gileshdhttps://github.com/gileshd
Jul 21, 2024https://github.com/probml/dynamax/pull/371#discussion_r1685772258
Please reload this pagehttps://github.com/probml/dynamax/pull/371
Learn morehttps://docs.github.com/articles/managing-disruptive-comments/#hiding-a-comment
Please reload this pagehttps://github.com/probml/dynamax/pull/371
https://github.com/gileshd
gileshdhttps://github.com/gileshd
Jul 21, 2024https://github.com/probml/dynamax/pull/371#event-13593103898
https://github.com/murphyk
murphykhttps://github.com/murphyk
Jul 22, 2024https://github.com/probml/dynamax/pull/371#issuecomment-2243283748
https://ott-jax.readthedocs.io/en/latest/_autosummary/ott.tools.k_means.k_means.htmlhttps://ott-jax.readthedocs.io/en/latest/_autosummary/ott.tools.k_means.k_means.html
https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.approx_max_k.htmlhttps://jax.readthedocs.io/en/latest/_autosummary/jax.lax.approx_max_k.html
Please reload this pagehttps://github.com/probml/dynamax/pull/371
https://github.co/hiddenchars
https://github.com/probml/dynamax/pull/{{ revealButtonHref }}
Sign up for freehttps://github.com/join?source=comment-repo
Sign in to commenthttps://github.com/login?return_to=https%3A%2F%2Fgithub.com%2Fprobml%2Fdynamax%2Fpull%2F371
https://github.com/slinderman
slinderman https://github.com/slinderman
https://github.com/gileshd
gileshd https://github.com/gileshd
Support for kmeans initialization with vmap https://github.com/probml/dynamax/issues/315
Please reload this pagehttps://github.com/probml/dynamax/pull/371
https://github.com/gileshd
https://github.com/murphyk
https://github.com
Termshttps://docs.github.com/site-policy/github-terms/github-terms-of-service
Privacyhttps://docs.github.com/site-policy/privacy-policies/github-privacy-statement
Securityhttps://github.com/security
Statushttps://www.githubstatus.com/
Communityhttps://github.community/
Docshttps://docs.github.com/
Contacthttps://support.github.com?tags=dotcom-footer

Viewport: width=device-width

URLs of crawlers that visited me.