jax

JAX를 이용한 미분 가능한 유한 요소법

Tevatron은 규모, 언어, 양식 전반에 걸쳐 통합된 문서 검색 툴킷입니다. SIGIR 2023 및 SIGIR 2025에서 데모되었습니다.

차량 번호판 텍스트 인식을 위한 가볍고 빠른 OCR 모델.

머신러닝 가이드. 머신러닝 도구, 라이브러리, 프레임워크, 거대 언어 모델(LLM) 및 학습 모델에 대해 모두 알아보세요.

다체 양자 시스템을 위한 머신러닝 알고리즘

물리 기반 머신러닝 튜토리얼 (Pytorch 및 Jax)

JAX 및 Equinox 기반의 가우시안 프로세스

JAX 기반의 벡터화된 강화학습 게임 환경

미분 가능한 유체 역학 패키지

JAX+Equinox 기반의 비선형 최적화(근 찾기, 최소 자승법 등). https://docs.kidger.site/optimistix/

JAX 및 Equinox 기반의 선형 솔버입니다.

FBPINN(Finite Basis Physics-Informed Neural Networks)을 사용하여 편미분 방정식과 관련된 순방향 및 역방향 문제를 해결합니다.

Google Cloud TPU에 대해 알아야 할 모든 정보입니다.

Felafax는 비 NVIDIA GPU를 위한 AI 인프라를 구축합니다

교사 모델에서 타일 모델까지 — 처음부터 구현한 LLM 증류 및 서빙 엔진: 커스텀 Triton/CUDA 커널, FSDP 증류, 페이징된 KV 연속 배치, 추측 디코딩, Rust 게이트웨이, JAX 오라클, 해석 가능성 도구를 포함합니다.

Pax는 대규모 모델 학습을 위한 Jax 기반 머신러닝 프레임워크입니다. 고급 구성 및 병렬 처리가 가능하며 업계 최고 수준의 모델 FLOP 활용률을 보여줍니다.

Python 기반 HPC 애플리케이션의 고속 처리를 위한 JAX 배열의 제로 카피 MPI 통신

Orbax는 JAX 사용자를 위한 공통 체크포인트 및 지속성 유틸리티를 제공합니다.

Yahoo Open-NSFW 모델의 Keras 구현체

https://srush.github.io/annotated-s4 구현체

단일 강체 모델을 기반으로 하고 Python으로 작성된 4족 보행 로봇용 모델 예측 제어기. 경사 기반(acados) 또는 샘플링 기반(jax)

XLB: 물리 기반 ML을 위한 가속 Lattice Boltzmann (XLB)