JAX is a key framework for LLM development, offering composable function transformations and a powerful bridge between low-level compilers and high-level code. To help address the challenges of moving from development to large-scale production, this talk introduces JAX-Toolbox, an open-source project that provides a robust foundation for the LLM development lifecycle. The session covers the CI/CD architecture that provides a stable foundation for JAX-based frameworks, how to build GPU-optimized containers for LLM frameworks such as MaxText and AXLearn to ensure reproducible workflows, and practical methods for deploying frameworks' containers on Kubernetes and SLURM-based clusters.
talk-data.com
Topic
ci/cd
1
tagged
Activity Trend
6
peak/qtr
2020-Q1
2026-Q1
Top Events
NYC Airflow Rooftop Happy Hour ft. PMC Member Jarek Potiuk!
2
9th July 25 | East Midlands Data | Nottingham | Speaker - Anna-Maria Wykes
1
Migrating to Bazel Without Breaking Things && Embedded Swift: A Practical Intro
1
Live Workshop: Advanced Kubernetes Platforms on Google Cloud with Pulumi
1
Build & Learn: Data Science with Coffee
1
From Simulation Algorithms to Production-Grade Data Systems
1
Berlin dbt Meetup
1
DevOps for Fabric - Git & Deployment Pipelines
1
Building a Test Automation Strategy That Scales | Meetup | QA, QE & Testers
1
@Viam: Optimizing CI/CD && Component Design Systems && Prototyping Flutter apps
1
DevOps Workshop: Get Started with CI/CD for AWS using Pulumi & GitHub Actions
1
How We Improve Developer Experience at Grammarly
1
Filtering by:
Stefano Bosisio
×