▲ 6 r/huggingface+2 crossposts

Hyper-transformer: Hybrid-Manifold Transformers with Hyperbolic Geometry and Spiking Neural Networks in PyTorch

Hey everyone,

Standard transformers operate strictly in flat Euclidean space with continuous activations. While effective, this creates challenges when modeling hierarchical tree-structured data and incurs significant energy overhead.

I built Hyper-transformer to explore a hybrid architecture integrating non-Euclidean geometry and neuromorphic spiking dynamics directly into transformer layers.

Key Technical Details:

• Hyperbolic Space Representation: Uses Poincaré and Lorentz manifold projections to capture hierarchical relationships with lower embedding dimensions.

• Spiking Neural Dynamics: Employs Leaky Integrate-and-Fire (LIF) spiking neurons for sparse, event-driven activation states.

• Hybrid Attention: Combines Riemannian distance metrics with attention mechanisms to handle mixed-geometry representations.

• Clean, modular PyTorch implementation designed for ablation studies.

I will drop the full GitHub repository link in the comments below! Would love to hear thoughts from anyone experimenting with non-Euclidean deep learning, manifold learning, or SNNs.

repo link : https://github.com/Griffith-7/Hyper-transformer.git

reddit.com
u/Griffith-07 — 2 days ago
▲ 7 r/CUDA+1 crossposts

H-JEPA-LM: Hierarchical Joint-Embedding Predictive Language Model in PyTorch

* Hey everyone, I wanted to share a research implementation I built focusing on non-generative, latent-space language modeling: **H-JEPA-LM**. Instead of predicting tokens autoregressively in surface vocabulary space, H-JEPA-LM leverages a Joint-Embedding Predictive Architecture (JEPA) to predict abstract representations directly in latent space. **Key Features:** • **Latent-Space Prediction:** Focuses on semantic representation learning rather than token generation. • **Hierarchical Abstraction:** Multi-level latent dynamics for both short-term syntax and long-term planning. • **Action-Conditioned Rollouts:** Supports world-model-style latent planning for downstream tasks. • Modular PyTorch implementation designed for experimentation. *I will drop the full GitHub repository link in the comments section below! I'd love to hear thoughts from anyone experimenting with JEPA, latent-space models, or world models.*

repo link : [https://github.com/Griffith-7/H-JEPA-LM.git\](https://github.com/Griffith-7/H-JEPA-LM.git)

reddit.com
u/Griffith-07 — 16 days ago
▲ 32 r/deeplearning+1 crossposts

H-JEPA-LM: Hierarchical Joint-Embedding Predictive Language Model in PyTorch

  • Hey everyone, I wanted to share a research implementation I built focusing on non-generative, latent-space language modeling: H-JEPA-LM. Instead of predicting tokens autoregressively in surface vocabulary space, H-JEPA-LM leverages a Joint-Embedding Predictive Architecture (JEPA) to predict abstract representations directly in latent space. Key Features:Latent-Space Prediction: Focuses on semantic representation learning rather than token generation. • Hierarchical Abstraction: Multi-level latent dynamics for both short-term syntax and long-term planning. • Action-Conditioned Rollouts: Supports world-model-style latent planning for downstream tasks. • Modular PyTorch implementation designed for experimentation. I will drop the full GitHub repository link in the comments section below! I'd love to hear thoughts from anyone experimenting with JEPA, latent-space models, or world models.

repo link : https://github.com/Griffith-7/H-JEPA-LM.git

reddit.com
u/Griffith-07 — 19 days ago

High-Performance C++20 Optical Neural Network (ONN) Simulator

  • Hey everyone, I wanted to share a project I've been developing focused on hardware-level AI acceleration: a high-performance C++20 simulator for Optical Neural Networks (ONNs).
  • Optical computing executes matrix multiplication using light interference through Mach-Zehnder Interferometer (MZI) meshes, offering extreme speed and efficiency, but simulating physical hardware constraints is critical before chip fabrication.
  • Key Features:MZI Mesh Architectures: Simulates Clements and Reck designs for unitary matrix transformations. • SVD Weight Mapping: Decomposes weight matrices into optical phase shifts. • Physical Error Modeling: Simulates thermal phase drift, laser power fluctuations, and photodetector shot noise. • Modern C++20 design leveraging Eigen3 for linear algebra.
  • I will drop the full GitHub repository link in the comments section below! I'd love to hear your thoughts on the numerical modeling approach or C++ optimization techniques.

repo link : https://github.com/Griffith-7/photonic-computing-simulation.git

reddit.com
u/Griffith-07 — 25 days ago
▲ 18 r/CUDA+4 crossposts

High-Performance C++20 Optical Neural Network (ONN) Simulator

Hey everyone,

I wanted to share a project I've been developing focused on hardware-level AI acceleration: a high-performance C++20 simulator for Optical Neural Networks (ONNs).

Optical computing executes matrix multiplication using light interference through Mach-Zehnder Interferometer (MZI) meshes, offering extreme speed and efficiency, but simulating physical hardware constraints is critical before chip fabrication.

Key Features:MZI Mesh Architectures: Simulates Clements and Reck designs for unitary matrix transformations. • SVD Weight Mapping: Decomposes weight matrices into optical phase shifts. • Physical Error Modeling: Simulates thermal phase drift, laser power fluctuations, and photodetector shot noise. • Modern C++20 design leveraging Eigen3 for linear algebra.

I will drop the full GitHub repository link in the comments section below! I'd love to hear your thoughts on the numerical modeling approach or C++ optimization techniques.

repo link : https://github.com/Griffith-7/photonic-computing-simulation.git

reddit.com
u/Griffith-07 — 24 days ago

Tpo-torch: Stable RLHF alignment in PyTorch using Target Policy Optimization

Hey everyone,

RLHF alignment using standard Proximal Policy Optimization (PPO) can be notoriously tricky to stabilize during LLM post-training due to policy collapse and high sensitivity to hyperparameters.

I built Tpo-torch to explore Target Policy Optimization (TPO) as a cleaner, more stable alternative for preference alignment directly in PyTorch.

Key Focus Areas:

• Mitigating policy collapse without requiring aggressive KL-divergence penalties.

• Modular, lightweight, and readable implementation designed for research and custom fine-tuning pipelines.

• Integrated stability benchmarks comparing policy drift against standard PPO.

I'll drop the GitHub repository link in the comments below! I'd love to hear feedback from anyone experimenting with alignment, preference optimization, or RLHF.

Repo link : https://github.com/Griffith-7/Tpo-torch.git

reddit.com
u/Griffith-07 — 29 days ago

Tpo-torch: Stable RLHF alignment in PyTorch using Target Policy Optimization

Hey everyone,

RLHF alignment using standard Proximal Policy Optimization (PPO) can be notoriously tricky to stabilize during LLM post-training due to policy collapse and high sensitivity to hyperparameters.

I built Tpo-torch to explore Target Policy Optimization (TPO) as a cleaner, more stable alternative for preference alignment directly in PyTorch.

Key Focus Areas:

• Mitigating policy collapse without requiring aggressive KL-divergence penalties.

• Modular, lightweight, and readable implementation designed for research and custom fine-tuning pipelines.

• Integrated stability benchmarks comparing policy drift against standard PPO.

I'll drop the GitHub repository link in the comments below! I'd love to hear feedback from anyone experimenting with alignment, preference optimization, or RLHF.

Repo link : [https://github.com/Griffith-7/Tpo-torch.git\](https://github.com/Griffith-7/Tpo-torch.git)

reddit.com
u/Griffith-07 — 30 days ago
▲ 3 r/LocalLLM+2 crossposts

Tpo-torch: Stable RLHF alignment in PyTorch using Target Policy Optimization

Hey everyone,

RLHF alignment using standard Proximal Policy Optimization (PPO) can be notoriously tricky to stabilize during LLM post-training due to policy collapse and high sensitivity to hyperparameters.

I built Tpo-torch to explore Target Policy Optimization (TPO) as a cleaner, more stable alternative for preference alignment directly in PyTorch.

Key Focus Areas:

• Mitigating policy collapse without requiring aggressive KL-divergence penalties.

• Modular, lightweight, and readable implementation designed for research and custom fine-tuning pipelines.

• Integrated stability benchmarks comparing policy drift against standard PPO.

I'll drop the GitHub repository link in the comments below! I'd love to hear feedback from anyone experimenting with alignment, preference optimization, or RLHF.

Repo link : https://github.com/Griffith-7/Tpo-torch.git

reddit.com
u/Griffith-07 — 30 days ago
▲ 8 r/CUDA+2 crossposts

TTA-Torch: Real-time, confidence-gated Test-Time Adaptation using dynamic LoRA updates

Hey everyone,

I’ve been working on a runtime adaptation framework for local models. While traditional fine-tuning locks a model's weights into place, Test-Time Adaptation (TTA) allows the model to adjust dynamically to incoming data streams.

I built TTA-Torch to bring real-time, confidence-gated TTA to LLMs using dynamic LoRA tracking in PyTorch.

Core Highlights:

• Confidence-gating mechanism to determine when adaptation is necessary.

• Low-overhead dynamic LoRA adjustments during the inference pass.

• Clean, modular PyTorch implementation.

I’d love to know if anyone here has experimented with test-time evaluation strategies for local setups, or if you have any feedback on handling memory/kv-cache overhead during dynamic steps!

Repo link: https://github.com/Griffith-7/TTA-Torch

u/Griffith-07 — 1 month ago
▲ 25 r/CUDA+2 crossposts

Custom NF4 Triton kernel achieving up to 1.41x dequantization speedup over bitsandbytes

Hey everyone,

I’ve been working on optimizing the memory overhead that comes with 4-bit inference. I wrote a custom NF4 dequantization kernel using Triton to see if I could eliminate the C++ dispatch bottlenecks found in current baselines.

🚀 Key Results:

• Up to 1.41x speedup compared to the standard bitsandbytes implementation across various tensor shapes.

• Written completely in Python/Triton, making it super easy to inspect, customize, or drop directly into your PyTorch compilation pipelines.

• Passes the Unsloth AI founding engineer challenge requirements (14/14 points).

I'd love to hear the community's feedback, especially if anyone wants to run their own benchmarks on different GPU architectures or suggest further optimization tricks!

Source code & full implementation:

https://github.com/Griffith-7/nf4-triton-kernel

u/Griffith-07 — 1 month ago