Imagen de Portada

Jasmine: A Simple, Performant and Scalable JAX-based World Modeling Codebase

Fuente: arXiv

Guardado en:

Detalles Bibliográficos
Autores principales:	Mahajan, Mihir, Nguyen, Alfred, Srambical, Franz, Bauer, Stefan
Formato:	Preprint
Publicado:	2025
Materias:	Machine Learning Artificial Intelligence
Acceso en línea:	Acceder al recurso
Etiquetas:	Agregar Etiqueta Sin Etiquetas, Sea el primero en etiquetar este registro!

Ejemplares similares

Confucius Code Agent: Scalable Agent Scaffolding for Real-World Codebases
por: Wong, Sherman, et al.
Publicado: (2025)

Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX
por: Bonnet, Clément, et al.
Publicado: (2023)

JaxARC: A High-Performance JAX-based Environment for Abstraction and Reasoning Research
por: Aadam, et al.
Publicado: (2026)

Simple, Good, Fast: Self-Supervised World Models Free of Baggage
por: Robine, Jan, et al.
Publicado: (2025)

laplax -- Laplace Approximations with JAX
por: Weber, Tobias, et al.
Publicado: (2025)

PuzzleJAX: A Benchmark for Reasoning and Learning
por: Earle, Sam, et al.
Publicado: (2025)

CAX: Cellular Automata Accelerated in JAX
por: Faldor, Maxence, et al.
Publicado: (2024)

minimax: Efficient Baselines for Autocurricula in JAX
por: Jiang, Minqi, et al.
Publicado: (2023)

A Simple and Scalable Representation for Graph Generation
por: Jang, Yunhui, et al.
Publicado: (2023)

NAVIX: Scaling MiniGrid Environments with JAX
por: Pignatelli, Eduardo, et al.
Publicado: (2024)

Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX
por: Nishimori, Soichiro, et al.
Publicado: (2026)

Uncertainty Quantification for Gradient-based Explanations in Neural Networks
por: Mulye, Mihir, et al.
Publicado: (2024)

G-Core: A Simple, Scalable and Balanced RLHF Trainer
por: Wu, Junyu, et al.
Publicado: (2025)

Learning Bug Context for PyTorch-to-JAX Translation with LLMs
por: Phan, Hung, et al.
Publicado: (2025)

Marginals Before Conditionals
por: Sahasrabudhe, Mihir
Publicado: (2026)

Chargax: A JAX Accelerated EV Charging Simulator
por: Ponse, Koen, et al.
Publicado: (2025)

Training Agents Inside of Scalable World Models
por: Hafner, Danijar, et al.
Publicado: (2025)

WeChat-YATT: A Scalable, Simple, Efficient, and Production Ready Training Library
por: Wu, Junyu, et al.
Publicado: (2025)

Simple and Scalable Strategies to Continually Pre-train Large Language Models
por: Ibrahim, Adam, et al.
Publicado: (2024)

Teleporter Theory: A General and Simple Approach for Modeling Cross-World Counterfactual Causality
por: Li, Jiangmeng, et al.
Publicado: (2024)

Higher Embedding Dimension Creates a Stronger World Model for a Simple Sorting Task
por: Bhalla, Brady, et al.
Publicado: (2025)

Soft $Q(λ)$: A multi-step off-policy method for entropy regularised reinforcement learning using eligibility traces
por: Mahajan, Pranav, et al.
Publicado: (2026)

CaRL: Learning Scalable Planning Policies with Simple Rewards
por: Jaeger, Bernhard, et al.
Publicado: (2025)

Scaling Is All You Need: Autonomous Driving with JAX-Accelerated Reinforcement Learning
por: Harmel, Moritz, et al.
Publicado: (2023)

Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization
por: Berasi, Davide, et al.
Publicado: (2026)

Consistency Models for Scalable and Fast Simulation-Based Inference
por: Schmitt, Marvin, et al.
Publicado: (2023)

JaxMARL: Multi-Agent RL Environments and Algorithms in JAX
por: Rutherford, Alexander, et al.
Publicado: (2023)

Scalable AI Inference: Performance Analysis and Optimization of AI Model Serving
por: Pham, Hung Cuong, et al.
Publicado: (2026)

Sanity Checks for Explanation Uncertainty
por: Valdenegro-Toro, Matias, et al.
Publicado: (2024)

JPC: Flexible Inference for Predictive Coding Networks in JAX
por: Innocenti, Francesco, et al.
Publicado: (2024)

Self-Questioning Language Models
por: Chen, Lili, et al.
Publicado: (2025)

Generalized Orders of Magnitude for Scalable, Parallel, High-Dynamic-Range Computation
por: Heinsen, Franz A., et al.
Publicado: (2025)

Future-as-Label: Scalable Supervision from Real-World Outcomes
por: Turtel, Benjamin, et al.
Publicado: (2026)

Learning Hidden Markov Models Using Conditional Samples
por: Kakade, Sham M., et al.
Publicado: (2023)

Amortized Active Causal Induction with Deep Reinforcement Learning
por: Annadani, Yashas, et al.
Publicado: (2024)

PyG 2.0: Scalable Learning on Real World Graphs
por: Fey, Matthias, et al.
Publicado: (2025)

M2Lingual: Enhancing Multilingual, Multi-Turn Instruction Alignment in Large Language Models
por: Maheshwary, Rishabh, et al.
Publicado: (2024)

Scalable Property Valuation Models via Graph-based Deep Learning
por: Riveros, Enrique, et al.
Publicado: (2024)

Better World Models Can Lead to Better Post-Training Performance
por: Gupta, Prakhar, et al.
Publicado: (2025)

An Embarrassingly Simple Approach to Enhance Transformer Performance in Genomic Selection for Crop Breeding
por: Chen, Renqi, et al.
Publicado: (2024)