Picture for Tal Ben-Nun

Tal Ben-Nun

Modeling Code: Is Text All You Need?

Add code
Jul 15, 2025
Viaarxiv icon

Leveraging AI for Productive and Trustworthy HPC Software: Challenges and Research Directions

Add code
May 13, 2025
Viaarxiv icon

BOOM: Benchmarking Out-Of-distribution Molecular Property Predictions of Machine Learning Models

Add code
May 03, 2025
Viaarxiv icon

Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training

Add code
Mar 24, 2025
Viaarxiv icon

Lion Cub: Minimizing Communication Overhead in Distributed Lion

Add code
Nov 25, 2024
Viaarxiv icon

VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores

Add code
Oct 03, 2023
Figure 1 for VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
Figure 2 for VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
Figure 3 for VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
Figure 4 for VENOM: A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
Viaarxiv icon

Cached Operator Reordering: A Unified View for Fast GNN Training

Add code
Aug 23, 2023
Viaarxiv icon

STen: Productive and Efficient Sparsity in PyTorch

Add code
Apr 15, 2023
Figure 1 for STen: Productive and Efficient Sparsity in PyTorch
Figure 2 for STen: Productive and Efficient Sparsity in PyTorch
Figure 3 for STen: Productive and Efficient Sparsity in PyTorch
Figure 4 for STen: Productive and Efficient Sparsity in PyTorch
Viaarxiv icon

Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization

Add code
Mar 14, 2023
Figure 1 for Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization
Figure 2 for Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization
Figure 3 for Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization
Figure 4 for Performance Embeddings: A Similarity-based Approach to Automatic Performance Optimization
Viaarxiv icon

A Theory of I/O-Efficient Sparse Neural Network Inference

Add code
Jan 03, 2023
Figure 1 for A Theory of I/O-Efficient Sparse Neural Network Inference
Figure 2 for A Theory of I/O-Efficient Sparse Neural Network Inference
Figure 3 for A Theory of I/O-Efficient Sparse Neural Network Inference
Figure 4 for A Theory of I/O-Efficient Sparse Neural Network Inference
Viaarxiv icon