jax

JAX による微分可能有限要素法

Tevatron - スケール、言語、モダリティを横断する統合ドキュメント検索ツールキット。SIGIR 2023、SIGIR 2025 デモ採択。

ナンバープレート認識のための軽量で高速な OCR モデル

機械学習ガイド。機械学習のツール、ライブラリ、フレームワーク、大規模言語モデル（LLMs）、モデルのトレーニングについて学ぶ

多体量子システム向けの機械学習アルゴリズム

物理情報機械学習チュートリアル（PyTorch および JAX）

JAXとEquinoxによるガウス過程

♟️ JAXによるベクトル化されたRLゲーム環境

微分可能流体力学パッケージ

JAX+Equinoxにおける非線形最適化（根の探索、最小二乗法など）。https://docs.kidger.site/optimistix/

JAX および Equinox における線形ソルバー。https://docs.kidger.site/lineax

有限基底物理情報ニューラルネットワーク（FBPINN）を使用した偏微分方程式に関連する順問題および逆問題の解決

Google Cloud TPUについて知りたいすべてのこと

Felafaxは、NVIDIA製以外のGPU向けAIインフラストラクチャを構築しています。

教師モデルからタイルまで — フルスクラッチのLLM蒸留・サービングエンジン：カスタムTriton/CUDAカーネル、FSDP蒸留、ページ化KV連続バッチ処理、推測デコーディング、Rustゲートウェイ、JAXオラクル、および解釈可能性ツール。

Pax は、大規模モデルのトレーニングのための Jax ベースの機械学習フレームワークです。高度かつ完全に設定可能な実験と並列処理を可能にし、業界をリードするモデルの FLOP 利用率を示しています。

Python の超高速 HPC アプリケーション向け、JAX 配列のゼロコピー MPI 通信 :zap:

Orbax は、JAX ユーザー向けに共通のチェックポイントおよび永続化ユーティリティを提供します

Yahoo Open-NSFW モデルの Keras 実装

https://srush.github.io/annotated-s4 の実装

単一剛体モデルに基づき Python で記述された、4足歩行ロボット用のモデル予測コントローラー。勾配ベース (acados) またはサンプリングベース (jax)

XLB: 物理ベースMLのための高速化格子ボルツマン法（XLB）