Functional JAX middleware using a compute Node abstraction
☆22Sep 6, 2026Updated last month
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:
- Automatically prove mathematical statements about your jax code in lean☆52Aug 6, 2026Updated 2 months ago
- A relay between Gymnasium and any software☆11Aug 11, 2025Updated last year
- ☆12Aug 5, 2026Updated 2 months ago
- Typescript implementation of Kinetix☆12Feb 25, 2025Updated last year
- Sheaf-ADMM☆76Jul 3, 2026Updated 3 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Megatron-LM adapted for Puro-2B☆100Aug 30, 2026Updated last month
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- PyTorch reproduction of DreamerV4 (Hafner et al., 2025): masked-autoencoder tokenizer, block-causal flow-matching dynamics with bootstrap…☆39Updated this week
- Single neuron simulator/ prediction benchmark; uses Hodgkin-Huxley with novel mechanisms☆47Aug 12, 2026Updated 2 months ago
- Live 3D viewer and editor for MuJoCo MJCF models, inside VS Code☆31Jul 22, 2026Updated 2 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Pretraining Recurrent Networks without Recurrence☆76Jun 7, 2026Updated 4 months ago
- Reversible molecular simulation☆14May 28, 2025Updated last year
- ☆37Oct 11, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Brax Viewer is a real-time, interactive web viewer for monitoring reinforcement learning (RL) policies☆17Nov 28, 2025Updated 10 months ago
- Code for the reproduction of counting manifolds☆16Feb 26, 2026Updated 7 months ago
- ☆22Nov 8, 2021Updated 4 years ago
- Code for the figures in Chapter 13 of "Reinforcement Learning: An Introduction" by Sutton and Barto☆14Jul 6, 2023Updated 3 years ago
- The Union of Intersections Framework in Python☆15Oct 7, 2026Updated last week
- Fiddling with Lindenmayer systems☆15Mar 24, 2023Updated 3 years ago
- Learning notes for understanding modern Machine Learning System.☆64Updated this week
- Official implementation of "NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flows"☆19Sep 22, 2025Updated last year
- PyTorch/PyG implementation of M3GNet☆12Sep 11, 2023Updated 3 years 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.
- ☆58Apr 22, 2026Updated 5 months ago
- Pseudopotential converter from upf to psp8☆11Jan 25, 2023Updated 3 years ago
- ☆19Mar 10, 2026Updated 7 months ago
- Fork of Flame repo for training of some new stuff in development☆20Aug 27, 2026Updated last month
- The official implementation for "ImageDoctor: Diagnosing Text-to-Image Generation via Grounded Image Reasoning"☆18Jul 31, 2026Updated 2 months ago
- ☆17Jul 23, 2026Updated 2 months ago
- Array quantization and compression☆14Dec 8, 2023Updated 2 years ago
- Main repo for the Boss API☆21Nov 21, 2024Updated last year
- A neurosymbolic T5 agent for playing text games, from the EACL 2023 paper "Behavior Cloned Transformers are Neurosymbolic Reasoners"☆21Feb 25, 2023Updated 3 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Sparse Transformers in JAX☆33Sep 17, 2026Updated last month
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- The AgentForge project focuses on building general tooling to construct multicapability AI systems by composing skills and models togethe…☆18Oct 11, 2023Updated 3 years ago
- Build a compiler to solve Anthropic's interview challenge.☆135Jul 12, 2026Updated 3 months ago
- ☆17Apr 22, 2025Updated last year
- Official repo for "Binary Retrieval-augmented Reward Mitigates Hallucinations"☆16Nov 13, 2025Updated 11 months ago
- Prangster tool from the Black Hat USA 2013 presentation "Black-Box Assessment of Pseudorandom Algorithms" by Derek Soeder, Christopher Ab…☆11Jul 9, 2014Updated 12 years ago