arXiv (All)
open access
E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU
We present e3j, a fast Euclid-equivariance backend for geometric deep learning applications with JAX bindings for GPU and TPU. Leveraging both optimized CUDA and Pallas kernels and algorithmic improvements, the library achieves state-of-the-art throughput and runtime on both forward and backward pat
Conceptio Open Archive
Analyze & explore related research in Conceptio
Connect this record directly into your research workflow or AI agent. Explore citations across open repositories, query full text, and export structured proof bundles.
This document is indexed with metadata only: full text is not available in the archive for this record. Open the official source ↗
Related documents
- E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU#1108702
- hyperbolix: Hyperbolic Deep Learning in JAX#1064792
- SakanaAI/AI-CUDA-Engineer-Archive#786764
- Making Room for AI: Multi-GPU Molecular Dynamics with Deep Potentials in GROMACS#2547
- AdamX: Cosine similarity meets gradient descent#999547
Record · ID 1129700
Retrieved via Conceptio: every document is proof-bundled with source, license, and retrieval metadata.
Retrieved via Conceptio: every document is proof-bundled with source, license, and retrieval metadata.
Bring this data into your agent: Connect via MCP or API · 20 free credits to start · no card required
Read the Docs → · View CLI source on GitHub · Conceptio Open Knowledge Archive
Read the Docs → · View CLI source on GitHub · Conceptio Open Knowledge Archive