Functional JAX middleware using a compute Node abstraction
☆22Sep 6, 2026Updated 2 months ago
Alternatives and similar repositories for nodejax
Users that are interested in nodejax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implicit Machine Learning Force Fields☆12Oct 12, 2026Updated 3 weeks ago
- Resources to learn llvm-ripple programming☆12Updated this week
- Save pytrees efficiently in HDF5 files.☆15Updated this week
- ☆15Jun 29, 2026Updated 4 months ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A relay between Gymnasium and any software☆11Aug 11, 2025Updated last year
- Code for the ICML 2023 paper "Learning to acquire novel cognitive tasks with evolution, plasticity and meta-meta-learning"☆11Jun 3, 2023Updated 3 years ago
- PyTorch operations for use in creating machine learning interatomic potentials☆22Sep 15, 2026Updated last month
- Official code base for la LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels extended to also learn late…☆15Apr 9, 2026Updated 6 months ago
- Developer blog for PyTorch☆23Oct 31, 2026Updated last week
- ☆22May 7, 2025Updated last year
- LES extension package for NequIP☆21Aug 14, 2026Updated 2 months ago
- JAX support for tvm-ffi abi☆28May 14, 2026Updated 5 months ago
- Sample-Efficient Online Learning in LM Agents via Hindsight Trajectory Rewriting☆14Jan 2, 2026Updated 10 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- minimal compiler☆24Feb 19, 2026Updated 8 months ago
- LUCID: Attention with Preconditioned Representations - A novel attention mechanism for improved long-context retrieval in Transformers☆13May 29, 2026Updated 5 months ago
- Virtual Decoupled Cores: Composable Programming Framework and Runtime for Async GPUs☆26Oct 27, 2026Updated last week
- Typescript implementation of Kinetix☆12Feb 25, 2025Updated last year
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- Go Wrappers for OpenXLA PJRT☆39Dec 12, 2025Updated 10 months ago
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated last year
- ☆12Oct 13, 2026Updated 3 weeks ago
- Brax Viewer is a real-time, interactive web viewer for monitoring reinforcement learning (RL) policies☆17Nov 28, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A Python module implemented in Rust for Phonopy☆11Updated this week
- ☆22Nov 8, 2021Updated 5 years ago
- Live 3D viewer and editor for MuJoCo MJCF models, inside VS Code☆31Jul 22, 2026Updated 3 months ago
- PyTorch/PyG implementation of M3GNet☆12Sep 11, 2023Updated 3 years ago
- Official implementation of "NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flows"☆19Sep 22, 2025Updated last year
- Fiddling with Lindenmayer systems☆15Mar 24, 2023Updated 3 years ago
- ☆19Jun 3, 2026Updated 5 months ago
- ☆58May 24, 2024Updated 2 years ago
- ☆10Jul 4, 2016Updated 10 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- High-performance CUDA kernels for equivariant graph neural networks (MACE, NequIP, Allegro). 10-20x faster than e3nn.☆44Jan 17, 2026Updated 9 months ago
- ☆19Mar 10, 2026Updated 7 months ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- A Julia Package for Active Inference☆42Jun 22, 2026Updated 4 months ago
- Partially Observable Multi-Agent RL with Transformers☆17Oct 22, 2026Updated 2 weeks ago
- Array quantization and compression☆14Dec 8, 2023Updated 2 years ago
- [MOVED → atlas-nano/codes/tree/main/2pt-legacy] Two phase approximation (solid + gas) for performing thermodynamic calculations on liquid…☆11Jul 20, 2026Updated 3 months ago