Buchumschlag

SoftJAX & SoftTorch: Empowering Automatic Differentiation Libraries with Informative Gradients

Fuente: arXiv

Gespeichert in:

Bibliographische Detailangaben
Hauptverfasser:	Paulus, Anselm, Geist, A. René, Musil, Vít, Hoffmann, Sebastian, Beker, Onur, Martius, Georg
Format:	Preprint
Veröffentlicht:	2026
Schlagworte:	Machine Learning
Online-Zugang:	Acceder al recurso
Tags:	Tag hinzufügen Keine Tags, Fügen Sie den ersten Tag hinzu!

Ähnliche Einträge

Differentiable Simulation of Hard Contacts with Soft Gradients for Learning and Control
von: Paulus, Anselm, et al.
Veröffentlicht: (2025)

LPGD: A General Framework for Backpropagation through Embedded Optimization Layers
von: Paulus, Anselm, et al.
Veröffentlicht: (2024)

Differentiation of Blackbox Combinatorial Solvers
von: Vlastelica, Marin, et al.
Veröffentlicht: (2019)

Novel Algorithms for Smoothly Differentiable and Efficiently Vectorizable Contact Manifold Construction
von: Beker, Onur, et al.
Veröffentlicht: (2026)

Optimizing Rank-based Metrics with Blackbox Differentiation
von: Rolínek, Michal, et al.
Veröffentlicht: (2019)

Deep Graph Matching via Blackbox Differentiation of Combinatorial Solvers
von: Rolínek, Michal, et al.
Veröffentlicht: (2020)

CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints
von: Paulus, Anselm, et al.
Veröffentlicht: (2021)

Backpropagation through Combinatorial Algorithms: Identity with Projection Works
von: Sahoo, Subham Sekhar, et al.
Veröffentlicht: (2022)

Smoothly Differentiable and Efficiently Vectorizable Contact Manifold Generation
von: Beker, Onur, et al.
Veröffentlicht: (2026)

Learning 3D-Gaussian Simulators from RGB Videos
von: Zhobro, Mikel, et al.
Veröffentlicht: (2025)

Soft Forward-Backward Representations for Zero-shot Reinforcement Learning with General Utilities
von: Bagatella, Marco, et al.
Veröffentlicht: (2026)

SoftDTW-CUDA-Torch: Memory-Efficient GPU-Accelerated Soft Dynamic Time Warping for PyTorch
von: Weber, Ron Shapira, et al.
Veröffentlicht: (2026)

Automatic Functional Differentiation in JAX
von: Lin, Min
Veröffentlicht: (2023)

Learning with 3D rotations, a hitchhiker's guide to SO(3)
von: Geist, A. René, et al.
Veröffentlicht: (2024)

Learning Bug Context for PyTorch-to-JAX Translation with LLMs
von: Phan, Hung, et al.
Veröffentlicht: (2025)

A Smooth Analytical Formulation of Collision Detection and Rigid Body Dynamics With Contact
von: Beker, Onur, et al.
Veröffentlicht: (2025)

Drifting Fields are not Conservative
von: Franz, Leonard T., et al.
Veröffentlicht: (2026)

Comparative Analysis of CNN Performance in Keras, PyTorch and JAX on PathMNIST
von: Nezović, Anida, et al.
Veröffentlicht: (2025)

Soft Decision Tree classifier: explainable and extendable PyTorch implementation
von: Shamir, Reuben R
Veröffentlicht: (2025)

Chunking the Critic: A Transformer-based Soft Actor-Critic with N-Step Returns
von: Tian, Dong, et al.
Veröffentlicht: (2025)

Colored Noise in PPO: Improved Exploration and Performance through Correlated Action Sampling
von: Hollenstein, Jakob, et al.
Veröffentlicht: (2023)

Directed Exploration in Reinforcement Learning from Linear Temporal Logic
von: Bagatella, Marco, et al.
Veröffentlicht: (2024)

DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
von: Rush, Keith, et al.
Veröffentlicht: (2024)

MPX: Mixed Precision Training for JAX
von: Gräfe, Alexander, et al.
Veröffentlicht: (2025)

jNO: A JAX Library for Neural Operator and Foundation Model Training
von: Armbruster, Leon, et al.
Veröffentlicht: (2026)

Soft Deterministic Policy Gradient with Gaussian Smoothing
von: Na, Hyunjun, et al.
Veröffentlicht: (2026)

Dual-Force: Enhanced Offline Diversity Maximization under Imitation Constraints
von: Kolev, Pavel, et al.
Veröffentlicht: (2025)

TorchResist: Open-Source Differentiable Resist Simulator
von: Wang, Zixiao, et al.
Veröffentlicht: (2025)

Lowering PyTorch's Memory Consumption for Selective Differentiation
von: Bhatia, Samarth, et al.
Veröffentlicht: (2024)

Pseudo-rigid body networks: learning interpretable deformable object dynamics from partial observations
von: Mamedov, Shamil, et al.
Veröffentlicht: (2023)

Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similarities
von: Zadaianchuk, Andrii, et al.
Veröffentlicht: (2023)

Beyond Exact Gradients: Convergence of Stochastic Soft-Max Policy Gradient Methods with Entropy Regularization
von: Ding, Yuhao, et al.
Veröffentlicht: (2021)

Active Fine-Tuning of Multi-Task Policies
von: Bagatella, Marco, et al.
Veröffentlicht: (2024)

Soft Hoeffding Tree: A Transparent and Differentiable Model on Data Streams
von: Köbschall, Kirsten, et al.
Veröffentlicht: (2024)

Soft-MSM: Differentiable Context-Aware Elastic Alignment for Time Series
von: Holder, Christopher, et al.
Veröffentlicht: (2026)

Soft Gradient Boosting with Learnable Feature Transforms for Sequential Regression
von: Karaca, Huseyin, et al.
Veröffentlicht: (2025)

Non-Stationary Learning of Neural Networks with Automatic Soft Parameter Reset
von: Galashov, Alexandre, et al.
Veröffentlicht: (2024)

JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
von: Toshev, Artur P., et al.
Veröffentlicht: (2024)

BlackJAX: Composable Bayesian inference in JAX
von: Cabezas, Alberto, et al.
Veröffentlicht: (2024)

jinns: a JAX Library for Physics-Informed Neural Networks
von: Gangloff, Hugo, et al.
Veröffentlicht: (2024)