Scholar4AI Updated 2026-11-03 14:43 UTC+09:00

jax-ml/jax: papers linking to it

GitHub · 36,568 stars · 4,016 forks · Python · license Apache-2.0 · created 2018 · github.com/jax-ml/jax

← Code and data repositories cited in research papers

Research papers in the 480 tracked journals that link to jax-ml/jax (https://github.com/jax-ml/jax), read from their Crossref records: by year, category and journal, the family tree of papers building on one another, and the linked repositories derived from it or that it derives from.

Share: X LinkedIn Bluesky Threads Facebook Reddit Hacker News Hatena LINE Email
Copy:

Key findings#

  • 28 papers in the tracked journals link to the GitHub repository jax-ml/jax in their Crossref records, the first in 2022; most of them (11) in 2025.
  • By category: Operations Research (14), AI (11) and Statistics (2); the journals with the most are Mathematical Programming (4), Nature Machine Intelligence (4) and Computational Optimization and Applications (3).
  • 2 of the papers build on one another: a later paper cites an earlier one that also links jax-ml/jax.

By year#

2022: 1 paper2023: 2 papers2024: 5 papers2025: 11 papers2026: 9 papers
20222026
Papers linking to jax-ml/jax by publication year
YearPapers
20221
20232
20245
202511
20269

Journals#

Family tree#

Each paper sits under the earlier paper it builds on; papers that cite no other paper here start a branch of their own.

Papers linking to jax-ml/jax, most cited first
PublishedTitleJournalAuthorsCitationsLink
2024-02-13A computational framework for neural network-based variational Monte Carlo with Forward LaplacianNature Machine IntelligenceRuichen Li, Haotian Ye, Du Jiang et al.40cites it
2022-05-19Deep autoencoders for acoustic anomaly detection: experiments with working machine and in-vehicle audioNeural Computing and ApplicationsGabriel Coelho, Luís Miguel Matos, Pedro José Pereira et al.26cites it
2025-06-06A multimodal conversational agent for DNA, RNA and protein tasksNature Machine IntelligenceBernardo P. de Almeida, Guillaume Richard, Hugo Dalla-Torre et al.26cites it
2023-01-17Majorization-minimization-based Levenberg–Marquardt method for constrained nonlinear least squaresComputational Optimization and ApplicationsNaoki Marumo, Takayuki Okuno, Akiko Takeda22cites it
2025-08-19Topic-aware influence maximization with deep reinforcement learning and graph attention networksData Mining and Knowledge DiscoveryTaha Halal, Bogdan Cautis, Benoît Groz et al.7cites it
2025-03-24Going faster to see further: graphics processing unit-accelerated value iteration and simulation for perishable inventory control using JAXAnnals of Operations ResearchJoseph Farrington, Wai Keong Wong, Kezhi Li et al.5cites it
2026-03-04Generalized policy improvement for efficient and robust multi-objective reinforcement learningAutonomous Agents and Multi-Agent SystemsLucas N. Alegre, Ana L. C. Bazzan, Diederik M. Roijers et al.4cites it
2023-07-12Complexity analysis of interior-point methods for second-order stationary points of nonlinear semidefinite optimization problemsComputational Optimization and ApplicationsShun Arahata, Takayuki Okuno, Akiko Takeda3cites it
2024-06-04Universal heavy-ball method for nonconvex optimization under Hölder continuous HessiansMathematical ProgrammingNaoki Marumo, Akiko Takeda3cites it
2025-11-19Empirical evaluation of normalizing flows in Markov chain Monte CarloMachine LearningDavid Nabergoj, Erik Štrumbelj3cites it
2026-03-25Machine learning global atomic representations with Euclidean fast attentionNature Machine IntelligenceJ. Thorben Frank, Stefan Chmiela, Klaus-Robert Müller et al.3cites it
2026-08-17A simultaneous approach for training neural differential-algebraic systems of equationsComputational Optimization and ApplicationsLaurens R. Lueg, Victor Manuel Cunha Alves, Daniel Schicksnus et al.3cites it
2024-07-04Efficient Shapley performance attribution for least-squares regressionStatistics and ComputingLogan Bell, Nikhil Devanathan, Stephen Boyd2cites it
2024-10-29SMT efficiency in supervised ML methods: a throughput and interference analysisJournal Of Big DataLucía Pons, Marta Navarro, Salvador Petit et al.2cites it
2024-11-04Accelerated-gradient-based generalized Levenberg–Marquardt method with oracle complexity bound and local quadratic convergenceMathematical ProgrammingNaoki Marumo, Takayuki Okuno, Akiko Takeda2cites it
2025-12-31Reusability report: Optimizing T count in general quantum circuits with AlphaTensor-QuantumNature Machine IntelligenceRemmy A. M. Zen, Maximilian Nägele, Florian Marquardt2cites it
2026-03-03Risk-sensitive actor-critic with static spectral risk measures for online and offline reinforcement learningExpert Systems with ApplicationsMehrdad Moghimi, Hyejin Ku2cites it
2026-08-10Condensed interior-point methods for scalable nonlinear programming on GPUsMathematical Programming ComputationFrançois Pacaud, Sungho Shin, Alexis Montoison et al.2cites it
2025-02-13CDOpt: a Python package for a class of Riemannian optimizationMathematical Programming ComputationNachuan Xiao, Xiaoyin Hu, Xin Liu et al.1cites it
2025-05-28Graph spring neural ODEs for link sign predictionMachine LearningAndrin Rehmann, Alexandre Bovet1cites it
2025-12-15Code generation for solving and differentiating through convex optimization problemsOptimization and EngineeringMaximilian Schaller, Stephen Boyd1cites it
2026-03-03Efficient inference in first passage time modelsStatistics and ComputingSicheng Liu, Alexander Fengler, Michael J. Frank et al.1cites it
2025-04-24Ozone: an open-source ordinary differential equation solver for gradient-based optimizationOptimization and EngineeringMark Z. Sperry, John Tan Teng Hwang0cites it
2025-08-13Error Bound Analysis of Physics-Informed Neural Networks for Solving Nonlinear Projection EquationsJournal of Optimization Theory and ApplicationsDawen Wu, Abdel Lisser0cites it
2025-10-01Transformation of Bi-level Transit Network Design Problem into Single-objective Unconstrained OptimizationNetworks and Spatial EconomicsJincheol Lee, Gyeongjun Kim, Keemin Sohn0cites it
2026-04-07Fast finite-sum optimization via cyclically-sampled Hessian averaging methodsMathematical ProgrammingThomas O’Leary-Roseberry, Raghu Bollapragada0cites it
2026-07-08Strongly convex maximization via the Frank-Wolfe algorithm with the Kurdyka-Łojasiewicz inequalityMathematical ProgrammingFatih S. Aktaş, Christian Kroer0cites it
2026-09-25Are farmers algorithm-averse? The case of decision support tools in crop managementFood PolicyAnna Massfeller, Daniel Hermann, Alexa Leyens et al.0cites it

How this is measured#

Links are read from each paper's Crossref record as its publisher deposited it: the URLs and repository DOIs in its reference list (the software and data it cites) and its supplementary links (“is-supplemented-by”: its own code or data). A paper that gives its code only in the text, a footnote or a data availability statement is not counted, and some publishers deposit reference lists without their URLs, so every count here is a lower bound. Repository details come from the GitHub REST API (stars, forks, and the repository a fork was made from), the Hugging Face Hub API (likes, downloads, and the base model a model was derived from) and DataCite (Zenodo and other data DOIs, and the GitHub release a Zenodo record archives), read again every month.

Cite this page

Scholar4AI. (2026). jax-ml/jax: papers linking to it. Journal and Conference Research Rankings. Retrieved November 3, 2026, from https://scholar4ai.com/code/github-jax-ml-jax

@misc{scholar4ai-code-github-jax-ml-jax,
  author = {{Scholar4AI}},
  title = {jax-ml/jax: papers linking to it},
  year = {2026},
  howpublished = {\url{https://scholar4ai.com/code/github-jax-ml-jax}},
  urldate = {2026-11-03},
  note = {Journal and Conference Research Rankings; data from OpenAlex (CC0); licensed CC BY-NC 4.0}
}

Free to reuse for non-commercial purposes with attribution (CC BY-NC 4.0).