All

7 repositories

simulacrax
Public
Physics-informed, RL-aligned evaluation engine for autonomous driving — JAX/Flax trajectory-diffusion models on the Waymo Open Dataset, with DPO alignment and p…
machine-learning reinforcement-learning deep-learning
machine-learning reinforcement-learning deep-learning evaluation flax autonomous-driving autonomous-vehicles trajectory-prediction self-driving-cars traffic-simulation
Python
•
MIT License
•0•0•0•0•Updated Aug 2, 2026Aug 2, 2026
artifex
Public
A research-focused modular generative modeling library built on JAX/Flax NNX
research generative-adversarial-network vae
research generative-adversarial-network vae gans multi-modal normalizing-flows variational-autoencoder generative-models diffusion-model geometric-models
Python
•
MIT License
•0•1•0•0•Updated Aug 2, 2026Aug 2, 2026
DiffBio
Public
End-to-end differentiable bioinformatics pipelines built on JAX/Flax NNX
machine-learning bioinformatics deep-learning
machine-learning bioinformatics deep-learning genomics protein-structure molecular-dynamics computational-biology rna-structure drug-discovery variant-calling
Python
•
MIT License
•0•1•0•0•Updated Jul 21, 2026Jul 21, 2026
calibrax
Public
Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical …
python benchmarking machine-learning
python benchmarking machine-learning deep-learning numpy ci-cd statistical-analysis scientific-computing flax profiling
Python
•
MIT License
•0•2•0•6•Updated Jul 20, 2026Jul 20, 2026
datarax
Public
A Differentiable Data Pipeline Framework for JAX
data-science data machine-learning
data-science data machine-learning jit autograd data-analysis differentiable jax xla flax-nnx
Python
•
MIT License
•0•2•0•0•Updated Jul 5, 2026Jul 5, 2026
opifex
Public
A unified scientific machine learning framework built on JAX/Flax NNX
jax scientific-machine-learning sciml
jax scientific-machine-learning sciml flax-nnx
Python
•
MIT License
•1•3•0•0•Updated Jun 24, 2026Jun 24, 2026
.github
Public
0•0•0•0•Updated May 5, 2026May 5, 2026

ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Avitai Bio

All

All

7 repositories

simulacrax

artifex

DiffBio

calibrax

datarax

opifex

.github

All

All

Repositories list

7 repositories