JAX

6 items

RESEARCHarXiv CS.AI·vor 1T

Accelerated Fourier SAT (AFSAT): Fully Realising a GPU-based Symmetric Pseudo-Boolean SAT Solver

Accelerated Fourier SAT (AFSAT) ist ein GPU-beschleunigter Solver für die pseudo-boolesche Erfüllbarkeit, der auf kontinuierlicher lokaler Suche basiert. Er verbessert die numerische Stabilität, Laufzeitleistung und Speichereffizienz im Vergleich zum Proof-of-Concept erheblich, indem er JAX für parallele Verarbeitung nutzt und Speicher-/Gleitkomma-Einschränkungen behebt.

SAT solver Computational logic GPU computing Optimization

DOC↑ trendingReddit r/MachineLearning·5/4/2026

Parax v0.5: Parametric Modeling in JAX [P]

Gary hat die Veröffentlichung von Parax v0.5 bekannt gegeben, einer generalisierten JAX-Bibliothek für parametrische Modellierung, die sich nun auf eine saubere, erweiterbare API und einen Opt-in-Ansatz konzentriert. Das Update führt Funktionen wie abgeleitete/eingeschränkte Parameter, berechnete PyTrees und Bearbeitungswerkzeuge ein, wobei die Dokumentation für Benutzer verfügbar ist.

software update AI tools Parametric Modeling JAX

NEWS↑ trendingReddit r/MachineLearning·4/9/2026

Parax: Parametric Modeling in JAX + Equinox [P]

Gary apresenta Parax, uma nova biblioteca Python construída sobre Equinox e JAX, projetada para aprimorar a modelagem paramétrica com metadados e manipulação de hierarquias de parâmetros profundas. A ferramenta visa oferecer uma abordagem mais orientada a objetos para inspeção e manipulação de modelos em aplicações científicas, mantendo os princípios de imutabilidade do Equinox.

Equinox machine learning Bibliotecas Python

ARTICLEDEV.to AI·5/1/2026

I Rebuilt Karpathy's NanoChat in JAX. Here's What XLA Gets Right and What It Gets Dead Wrong.

Dieser Inhalt beschreibt die Portierung von Andrej Karpathys NanoChat von PyTorch auf JAX/Flax NNX, wodurch schnelles Training auf einer einzelnen GPU und TPU-Kompatibilität erreicht wurde. Er erläutert die Vorteile von XLA bei der Eliminierung des Python-Overheads, hebt aber auch dessen Einschränkungen bei fortschrittlichen Funktionen und beim Debugging hervor.

deep learning XLA JAX PyTorch

RESEARCHarXiv CS.AI·vor 19T

Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX

Mahjax ist eine neue, vollständig vektorisierte Riichi Mahjong-Umgebung, implementiert in JAX, die eine großflächige Rollout-Parallelisierung auf GPUs für die Forschung im Bereich des Reinforcement Learnings ermöglicht. Sie erleichtert das Tabula-rasa-Lernen und enthält ein hochwertiges Visualisierungstool zur Fehlerbehebung und Interaktion mit trainierten Agenten.

reinforcement learning learning GPU Mahjong

DOCGoogle for Developers (YouTube)·4/30/2026

Unlocking Low-Level Control: Customizing Keras Training Loops with JAX

Dieser Inhalt behandelt, wie man Low-Level-Kontrolle erlangt und Keras-Trainingsschleifen anpasst. Er beschreibt die Integration mit JAX, um größere Flexibilität und Leistung bei der Entwicklung von Machine-Learning-Modellen zu ermöglichen.

Training Loops Keras deep learning machine learning

Unlocking Low-Level Control: Customizing Keras Training Loops with JAX