Pinned Loading

1. the-modern-transformer the-modern-transformer Public

   Python

2. llm-reasoning-jax llm-reasoning-jax Public

   Post-Training Alignment and Reinforcement Learning for Reasoning LLMs in JAX.

   Python 1 1

3. proxy-kd-jax proxy-kd-jax Public

   JAX implementation of "Knowledge Distillation of Black-Box Large Language Models" (Chen et al., 2024)

   Python

4. latent-diffusion-jax latent-diffusion-jax Public

   (WIP) A Latent Diffusion Model (Stable Diffusion) Implementation in JAX/Flax. Denoising Scheduler, UNet Architecture with text conditioning, and Classifier-Free Guidance from scratch.

   Python 1 1

5. hybrid-atp hybrid-atp Public

   A Neuro-Symbolic Automated Theorem Prover (ATP) bridging LLM intuition with Lean 4's formal verification. Features MCTS/DFS search and a Generator-Validator loop.

   Jupyter Notebook 1

6. dapo-jax dapo-jax Public

   Flax NNX implementation of Decoupled Clip and Dynamic Sampling Policy Optimization (DAPO) (what an acronym)

   Python